knowledgator
/

comprehend_it-base

Zero-Shot Classification

text-classification

information-extraction

Model card Files Files and versions

Ihor commited on Mar 24

Commit

93ca760

·

verified ·

1 Parent(s): 84c22d7

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -128,7 +128,7 @@ from transformers import TrainingArguments, Trainer
 backbone_model = AutoModelForSequenceClassification.from_pretrained('knowledgator/comprehend_it-base')
-loss_func = FocalLoss(multi_target=True)
 model = LiqFitModel(backbone_model.config, backbone_model, loss_func=loss_func)

 backbone_model = AutoModelForSequenceClassification.from_pretrained('knowledgator/comprehend_it-base')
+loss_func = FocalLoss()
 model = LiqFitModel(backbone_model.config, backbone_model, loss_func=loss_func)