andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Deployed f8db199 with MkDocs version: 1.3.1

Browse source
This commit is contained in:
2022-07-29 11:07:11 +00:00
parent 8a093ac9d2
commit d430336c47
11 changed files with 741 additions and 722 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/scibert/train/train.ipynb
View file

@ -246,7 +246,7 @@
" Dataset.from_dict({\"text\": X, \"label\": y})\n",
" .map(lambda v: tokenizer(v[\"text\"], truncation=True), batched=True)\n",
" .remove_columns(\"text\")\n",
" .train_test_split(test_size=0.2, shuffle=True)\n",
" .train_test_split(test_size=0.2, shuffle=True) # test is actually validation\n",
")\n",
"\n",
"f1_score = load_metric(\"f1\")\n",