Merge pull request #252 from SunYanCN/master

Update text-classification.md
BrikerMan · Sep 27, 2019 · 9e6d268 · 9e6d268
2 parents ee09e2e + 9ccb5dc
commit 9e6d268
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mkdocs/docs/tutorial/text-classification.md b/mkdocs/docs/tutorial/text-classification.md
@@ -79,7 +79,7 @@ import logging
 logging.basicConfig(level='DEBUG')
 
 bert_embed = BERTEmbedding('<PRE_TRAINED_BERT_MODEL_FOLDER>',
-                           task=kashgari.LABELING,
+                           task=kashgari.CLASSIFICATION,
                            sequence_length=100)
 model = BiGRU_Model(bert_embed)
 model.fit(train_x, train_y, valid_x, valid_y)