Closed shahid-cloud closed 3 years ago
Describe the bug I have successfully implemented my own multilabel classification notebook with 30 labels per your blog: https://towardsdatascience.com/multi-label-classification-using-bert-roberta-xlnet-xlm-and-distilbert-with-simple-transformers-b3e0cda12ce5
Now, I want to be able to get a sklearn classification_report as per the instructions here https://simpletransformers.ai/docs/tips-and-tricks/
However, when I run
`model = MultiLabelClassificationModel('roberta', 'roberta-large', num_labels=30, args=model_args)
model.train_model(train_df=train_df, eval_df=eval_df, classification_report=sklearn.metrics.classification_report)`
I get the following error:
ValueError Traceback (most recent call last)
This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.
Describe the bug I have successfully implemented my own multilabel classification notebook with 30 labels per your blog: https://towardsdatascience.com/multi-label-classification-using-bert-roberta-xlnet-xlm-and-distilbert-with-simple-transformers-b3e0cda12ce5
Now, I want to be able to get a sklearn classification_report as per the instructions here https://simpletransformers.ai/docs/tips-and-tricks/
However, when I run
`model = MultiLabelClassificationModel('roberta', 'roberta-large', num_labels=30, args=model_args)
Train the model
model.train_model(train_df=train_df, eval_df=eval_df, classification_report=sklearn.metrics.classification_report)`
I get the following error:
`
ValueError Traceback (most recent call last)