alihan/TurkishBERTweet
1
0
Fork 0
mirror of https://github.com/ViralLab/TurkishBERTweet.git synced 2023-12-19 18:19:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

refactored

Browse Source
This commit is contained in:
AliNajafi
2023-10-02 20:41:50 +03:00
parent e5e2a72e81
commit 7e0b76654a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@@ -96,7 +96,7 @@ if getattr(tokenizer, "pad_token_id") is None:
id2label_sa = {0: "negative", 2: "positive", 1: "neutral"}
turkishBERTweet_sa = AutoModelForSequenceClassification.from_pretrained(
peft_config.base_model_name_or_path, return_dict=True, num_labels=3, id2label=id2label_sa
peft_config.base_model_name_or_path, return_dict=True, num_labels=len(id2label_sa), id2label=id2label_sa
)
turkishBERTweet_sa = PeftModel.from_pretrained(turkishBERTweet_sa, peft_model)
@@ -148,7 +148,7 @@ if getattr(tokenizer, "pad_token_id") is None:
id2label_hs = {0: "No", 1: "Yes"}
turkishBERTweet_hs = AutoModelForSequenceClassification.from_pretrained(
peft_config.base_model_name_or_path, return_dict=True, num_labels=2, id2label=id2label_hs
peft_config.base_model_name_or_path, return_dict=True, num_labels=len(id2label_hs), id2label=id2label_hs
)
turkishBERTweet_hs = PeftModel.from_pretrained(turkishBERTweet_hs, peft_model)