This commit is contained in:
RubyZ10 2024-06-14 09:40:32 -07:00
Родитель b34a0c67e3
Коммит 626fba3386
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -128,7 +128,7 @@
"source": [
"def load_dataset(split):\n",
" config_kwargs = {\"name\": \"title_genre_classifiction\"}\n",
" dataset = datasets.load_dataset(\"blbooksgenre\", split=split, **config_kwargs)\n",
" dataset = datasets.load_dataset(\"blbooksgenre\", split=split, **config_kwargs, trust_remote_code=True)\n",
" return pd.DataFrame({\"text\": dataset[\"title\"], \"label\": dataset[\"label\"]})\n",
"\n",
"\n",