We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
書籍ではBertModelを利用した自作モデルを実装していますが、2023年現在ではBertForSequenceClassificationにてproblem_type="multi_label_classification"を指定するとマルチラベル分類ができるようです。
BertForSequenceClassification
problem_type="multi_label_classification"
self.bert_scml = BertForSequenceClassification.from_pretrained( model_name, num_labels=num_labels, problem_type="multi_label_classification" )
The text was updated successfully, but these errors were encountered:
ありがとうございます!こちら参考にさせていただきます。
Sorry, something went wrong.
No branches or pull requests
書籍ではBertModelを利用した自作モデルを実装していますが、2023年現在では
BertForSequenceClassification
にてproblem_type="multi_label_classification"
を指定するとマルチラベル分類ができるようです。The text was updated successfully, but these errors were encountered: