Skip to content

Commit fcd729f

Browse files
committed
🐛 remove unnecessary codes
1 parent 0d305e5 commit fcd729f

1 file changed

Lines changed: 1 addition & 12 deletions

File tree

scripts/NSMC/naver_review_classifications_pytorch_kobert.ipynb

Lines changed: 1 addition & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -12,16 +12,6 @@
1212
"metadata": {},
1313
"source": []
1414
},
15-
{
16-
"cell_type": "code",
17-
"execution_count": null,
18-
"metadata": {},
19-
"outputs": [],
20-
"source": [
21-
"!pip install ipywidgets # for vscode\n",
22-
"!pip install git+https://git@github.com/SKTBrain/KoBERT.git@master"
23-
]
24-
},
2515
{
2616
"cell_type": "code",
2717
"execution_count": null,
@@ -52,7 +42,7 @@
5242
"outputs": [],
5343
"source": [
5444
"# NSMC 영화 리뷰 데이터셋 준비\n",
55-
"dataset = load_dataset(\"e9t/nsmc\", trust_remote_code=True)\n",
45+
"dataset = load_dataset(\"e9t/nsmc\")\n",
5646
"\n",
5747
"def tokenize_function(example):\n",
5848
" return tokenizer(example[\"document\"])\n",
@@ -90,7 +80,6 @@
9080
" load_best_model_at_end=True, # 훈련 종료 시 가장 좋은 모델 로드\n",
9181
" metric_for_best_model=\"accuracy\", # 가장 좋은 모델을 선택하기 위한 메트릭\n",
9282
" push_to_hub=False, # Hugging Face Hub에 모델 업로드 여부\n",
93-
" max_steps=10,\n",
9483
")\n",
9584
"\n",
9685
"def compute_metrics(eval_pred):\n",

0 commit comments

Comments
 (0)