Ejemplo n.º 1
0
 class Config(DocumentClassificationTask.Config):
     model: NewBertModel.Config = NewBertModel.Config(
         inputs=NewBertModel.Config.BertModelInput(
             tokens=BERTTensorizer.Config(columns=["text1", "text2"],
                                          max_seq_len=128)))
     metric_reporter: ClassificationMetricReporter.Config = (
         ClassificationMetricReporter.Config(
             text_column_names=["text1", "text2"]))
Ejemplo n.º 2
0
 class Config(DocumentClassificationTask.Config):
     model: NewBertModel.Config = NewBertModel.Config()