Update sentiment.py

hiworldwzj · Jun 17, 2019 · 244645d · 244645d
1 parent 239e206
commit 244645d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sentiment.py b/sentiment.py
@@ -102,7 +102,7 @@ def __iter__(self):
 x2_in = Input(shape=(None,))
 
 x = bert_model([x1_in, x2_in])
-x = Lambda(lambda x: x[:, 0])(x)
+x = Lambda(lambda x: x[:, 0])(x) # 取出[CLS]对应的向量用来做分类
 p = Dense(1, activation='sigmoid')(x)
 
 model = Model([x1_in, x2_in], p)