update cnn model

captain · captain · commit 2e9b7e628175 · 2017-12-17T11:06:25.000+08:00
diff --git a/text_analysis/cnn_model.py b/text_analysis/cnn_model.py
@@ -37,7 +37,7 @@ def __init__(self, output_dimesion, vocab_size, dropout_rate, emb_dim, max_len,
         )
 
 
-        self.conv_1 =
+        self.conv_1 = nn.Conv2d(1, nb_filters, kernel_size=(3, emb_dim))
         self.conv_2 = nn.Conv2d(1, nb_filters, kernel_size=(4, emb_dim))
         self.conv_3 = nn.Conv2d(1, nb_filters, kernel_size=(5, emb_dim))
 

Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,7 @@ def __init__(self, output_dimesion, vocab_size, dropout_rate, emb_dim, max_len,`
`37`	`37`	`)`
`38`	`38`
`39`	`39`
`40`		`- self.conv_1 =`
	`40`	`+ self.conv_1 = nn.Conv2d(1, nb_filters, kernel_size=(3, emb_dim))`
`41`	`41`	`self.conv_2 = nn.Conv2d(1, nb_filters, kernel_size=(4, emb_dim))`
`42`	`42`	`self.conv_3 = nn.Conv2d(1, nb_filters, kernel_size=(5, emb_dim))`
`43`	`43`