fixed bad bert_sentence componen info
@@ -6,13 +6,13 @@ class BertSentence:
def get_default_model():
return BertSentenceEmbeddings.pretrained() \
.setInputCols("sentence") \
- .setOutputCol("bert_sentence")
+ .setOutputCol("sentence_embeddings")
@staticmethod
def get_pretrained_model(name, language, bucket=None):
return BertSentenceEmbeddings.pretrained(name,language,bucket) \
.setInputCols('sentence') \
- .setOutputCol("bert_sentence")
+ .setOutputCol("sentence_embeddings")
