search

stockmarkteam / bert-book

「BERTによる自然言語処理入門: Transformersを使った実践プログラミング」サポートページ
MIT License
259 stars 80 forks source link

第7章のマルチラベル分類のBERTモデルについて #56

Open kassy11 opened 1 year ago

kassy11 commented 1 year ago

書籍ではBertModelを利用した自作モデルを実装していますが、2023年現在ではBertForSequenceClassificationにてproblem_type="multi_label_classification"を指定するとマルチラベル分類ができるようです。

self.bert_scml = BertForSequenceClassification.from_pretrained(
    model_name, num_labels=num_labels, problem_type="multi_label_classification"
)
omitakahiro commented 1 year ago

ありがとうございます！こちら参考にさせていただきます。