Skip to content

Commit 696201c

Browse files
committed
parameter fix
1 parent 8ecf304 commit 696201c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

senti/senti_models.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -367,7 +367,7 @@ def fit_multiview_cnn_word_cnn_char(self):
367367
), CNNChar(
368368
batch_size=128, emb_X=emb_char.X, input_size=140, output_size=3, static_mode=1, f1_classes=[0, 2]
369369
)], output_size=3)
370-
kw = dict(val_docs=ft.transform(self.val_docs), val_y=self.val_labels(), average_classes=[0, 2])
370+
kw = dict(val_docs=ft.transform(self.val_docs), val_y=self.val_labels())
371371
cf.fit(ft.transform(distant_docs), distant_labels(), epoch_size=10**4, max_epochs=100, **kw)
372372
cf.fit(ft_typo.transform(self.train_docs), self.train_labels(), max_epochs=10, **kw)
373373
estimator = Pipeline([('features', ft), ('classifier', cf)])

0 commit comments

Comments
 (0)