|
|
@@ -16,8 +16,6 @@ class SbertForSentimentClassification(SbertForSequenceClassificationBase): |
|
|
|
|
|
|
|
Args: |
|
|
|
model_dir (str): the model path. |
|
|
|
model_cls (Optional[Any], optional): model loader, if None, use the |
|
|
|
default loader to load model weights, by default None. |
|
|
|
""" |
|
|
|
super().__init__( |
|
|
|
model_dir, *args, model_args={'num_labels': 2}, **kwargs) |
|
|
|