mosievich.kirill
commited on
Commit
·
2c26172
1
Parent(s):
779c97b
task_4 baseline
Browse files- ML2_2025_nlp_ops1.ipynb +188 -0
- app.py +73 -0
- config.json +338 -0
- model_train.ipynb +0 -0
- pytorch_model.bin +3 -0
- requirements.txt +3 -0
ML2_2025_nlp_ops1.ipynb
ADDED
|
@@ -0,0 +1,188 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cells": [
|
| 3 |
+
{
|
| 4 |
+
"cell_type": "markdown",
|
| 5 |
+
"metadata": {},
|
| 6 |
+
"source": [
|
| 7 |
+
"# __Девопсная домашка по трансформерам__\n",
|
| 8 |
+
"\n",
|
| 9 |
+
"## __Описание__\n",
|
| 10 |
+
"\n",
|
| 11 |
+
"\n",
|
| 12 |
+
"\n",
|
| 13 |
+
"Ваш главный квест на эту домашку - сделать свой простой сервис на трансформерах. Вот прям целый сервис: начиная с данных и заканчивая графическим интерфейсом где-то в интернете. Ваш сервис может решать либо одну из предложенных ниже задач, либо любую другую (что-то более дорогое лично вам).\n",
|
| 14 |
+
"\n",
|
| 15 |
+
"__Стандартная задача: классификатор статей.__ Нужно построить сервис который принимает название статьи и её abstract, и выдаёт наиболее вероятную тематику статьи: скажем, физика, биология или computer science. В интерфейсе должно быть можно ввести отдельно abstract, отдельно название -- и увидеть топ-95%* тематик, отсортированных по убыванию вероятности. Если abstract не ввели, нужно классифицировать статью только по названию. Ниже вас ждут инструкции и данные именно для этой задачи.\n",
|
| 16 |
+
"\n",
|
| 17 |
+
"<details><summary><u> Что значит Топ-95%?</u></summary>\n",
|
| 18 |
+
" Нужно выдавать темы по убыванию вероятности, пока их суммарная вероятность не превысит 95%. В зависимости от предсказанной вероятности, это может быть одна или более тем. Например, если модель предсказала вероятности [4%, 20%, 60%, 2%, 14%], нужно вывести 3 топ-3 класса. Если один из классов имеет вероятность 96%, достаточно вывести один этот класс.\n",
|
| 19 |
+
"</details>\n",
|
| 20 |
+
"\n",
|
| 21 |
+
"Альтернативно, вы можете отважиться сделать что-то своё, на данных из интернета или своих собственных. В вашей задаче обязательно должно быть _оправданное_ использование трансформеров. Использовать ML чтобы переводить часовые пояса - плохой план.\n",
|
| 22 |
+
"\n",
|
| 23 |
+
"Achtung: трансформеры круты, но не всемогущи. Далеко не любую задачу можно решить ощутимо лучше рандома. Для калибровки, вот несколько примеров решаемых задач (всё кликабельно):\n",
|
| 24 |
+
"\n",
|
| 25 |
+
"\n",
|
| 26 |
+
"<details><summary> - <b>[medium]</b> <u>Сгенерировать youtube-комментарии по _ссылке_ на видео</u></summary>\n",
|
| 27 |
+
" Всё просто, юзер постит ссылку на видео - вы его комментируете. Можно заранее обусловиться что видео только на английском или на русском. Нужно сочинить _несколько_ комментариев. Kudos если вместе с основным комментарием вы порождаете юзернеймы и-или ответы на него.\n",
|
| 28 |
+
" \n",
|
| 29 |
+
" Датасет для файнтюна можно [взять с kaggle](https://www.kaggle.com/tanmay111/youtube-comments-sentiment-analysis/data?select=UScomments.csv) или [собрать самостоятельно](https://towardsdatascience.com/how-to-build-your-own-dataset-of-youtube-comments-39a1e57aade).\n",
|
| 30 |
+
" \n",
|
| 31 |
+
" В качестве основной модели можно использовать [GPT-2 large](https://huggingface.co/gpt2-large). Вот как её файнтюнить: https://tinyurl.com/gpt2-finetune-colab . Если хотите больше - можно взять что-то из творчества https://huggingface.co/EleutherAI . Например, вот [тут](https://tinyurl.com/gpt-j-8bit) есть пример как файнтюнить GPT-J-6B (в 8 раз больше gpt2-large). Однако, этим стоит заниматься уже после того, как у вас заработал базовый сценарий с GPT2-large или даже base.\n",
|
| 32 |
+
" \n",
|
| 33 |
+
" В итоговом сервисе ��ожно дать пользователю вариировать параметры генерации: температура или top-p, если сэмплинг; beam size и length penalty, если beam search; сколько комментариев сгенерировать, etc. Отдельный респект если ваш код будет выводить комментарий по одному слову, прямо в процессе генерёжки - чтобы пользователь не ждал пока вы настругаете абзац целиком.\n",
|
| 34 |
+
"</details>\n",
|
| 35 |
+
"\n",
|
| 36 |
+
"<details><summary> - <b>[medium]</b> <u>Предсказать зарплату по профилю (симулятор Дудя).</u></summary>\n",
|
| 37 |
+
" Note: <details> <summary>Причём тут Дудь?</summary> <img src=https://www.meme-arsenal.com/memes/6dd85f126bbab4f9774ced71ffadbcb3.jpg> </details>\n",
|
| 38 |
+
" \n",
|
| 39 |
+
" Главная сложность задачи - достать хорошие данные. Если хороших данных не случилось - можно и трешовые :) Задание всё-таки про технологии а не про продукт. Для начала можно взять подмножество фичей [отсюда](https://www.kaggle.com/c/job-salary-prediction/data), которые вы можете восстановить из профиля linkedin - название профессии и компании. Название компании лучше заменить на фичи из открытых источников: сфера деятельности, размер, етц.\n",
|
| 40 |
+
" \n",
|
| 41 |
+
" А дальше файнтюним на этом BERT / T5 и радуемся. Ну или хотя бы смеёмся.\n",
|
| 42 |
+
"</details>\n",
|
| 43 |
+
"\n",
|
| 44 |
+
"\n",
|
| 45 |
+
"<details><summary> - <b>[hard]</b> <u>Мнения с географической окраской.</u></summary>\n",
|
| 46 |
+
" \n",
|
| 47 |
+
" Сервис который принимает на вход тему (хэштег или ключевую фразу) и рисует карту мира, где в каждом регионе показано, с какой эмоциональной окраской о ней высказываются в социальных сетях. В качестве социальной сети можно взять VK/twitter, в случая VK ожидается детализация не по странам, а по городам стран бывшего СССР.\n",
|
| 48 |
+
" \n",
|
| 49 |
+
" В минимальном варианте достаточно определять тональность твита в режиме \"позитивно-негативно\", зафайнтюнив условный BERT/T5 на одном из десятков {vk/twitter} sentiment classification датасетах. Географическую привязку можно получить из профиля пользователя. А дальше осталось собрать данные по странам и регионам.\n",
|
| 50 |
+
"\n",
|
| 51 |
+
"</details>\n",
|
| 52 |
+
"\n",
|
| 53 |
+
"\n",
|
| 54 |
+
"<details><summary> - <b>[very hard]</b> <u>Найти статью википедии по фото предмета статьи</u></summary>\n",
|
| 55 |
+
"\n",
|
| 56 |
+
" Чтобы можно было сфотать какую-нибудь неведомую чешуйню на телефон и получить сумму человеческих знаний о ней в форме вики-статьи.\n",
|
| 57 |
+
" \n",
|
| 58 |
+
" В качестве функции потерь можно использовать contrastive loss. Этот лосс неплохо описан в статье [CLIP](https://arxiv.org/abs/2103.00020). Вместо обучения с нуля предлагается взять, собственно, CLIP (text transformer + image transformer) отсюда: https://huggingface.co/docs/transformers/model_doc/clip. Модель будет сопоставлять каждой статьи и \n",
|
| 59 |
+
" \n",
|
| 60 |
+
" Данные для этого квеста можно собрать через API википедии: вики-статьи о предметах обычно содержит фото этого объекта и, собственно, текст статьи. Советуем собрать как минимум 10^4 пар картинка-статья. Картинки советуем дополнительно аугментировать как минимум стандартными картиночными аугами, как максимум - поиском похожих картинок в интернете / imagenet-е по тому же CLIP image encoder-у, но с исходными весами.\n",
|
| 61 |
+
" \n",
|
| 62 |
+
" На время отладки интерфейса рекомендуем ограничить��я небольшим списком статьей: условно, кошечки, собачки, птички, гаечные ключи, машины. Как станет понятно что оно работает \"на кошках\", можно расширить этот список до \"всех статей таких-то категорий\". Эмбединги статей лучше предпосчитать в файл. Если долго их перебирать - можно (но необязательно) воспользоваться быстрым поиском соседей, e.g. [faiss](https://github.com/facebookresearch/faiss) HNSW.\n",
|
| 63 |
+
"</details>\n",
|
| 64 |
+
"\n",
|
| 65 |
+
"\n",
|
| 66 |
+
"## __Как научить классификатор статей?__\n",
|
| 67 |
+
"\n",
|
| 68 |
+
"Данные для классификации статей можно скачать, например, [отсюда](https://www.kaggle.com/neelshah18/arxivdataset/). В этих данных есть заголовок и abstract статьи, а ещё поле __\"tag\"__: тематика статьи [по таксономии arxiv.org](https://arxiv.org/category_taxonomy). Вы можете расширить выборку, добавив в неё статьи за 2019-н.в. годы. Для этого можно [использовать arxiv API](https://github.com/lukasschwab/arxiv.py), самостоятельно распарсить arxiv с помощью [beautifulsoup](https://pypi.org/project/beautifulsoup4/), или поискать другие датасеты на kaggle, huggingface, etc.\n",
|
| 69 |
+
"\n",
|
| 70 |
+
"Когда данные собраны (и аккуратно нарезаны на train/test), можно что-нибудь и обучить. Мы советуем использовать для этого библиотеку `transformers`. Советуем, но не заставляем: если хочется, можно взять [fairseq roberta](https://github.com/pytorch/fairseq/blob/main/examples/roberta), [google t5](https://github.com/google-research/text-to-text-transfer-transformer) или даже написать всё с нуля.\n",
|
| 71 |
+
"\n",
|
| 72 |
+
"Мы разбирали transformers на [семинаре](https://lk.yandexdataschool.ru/courses/2025-spring/7.1332-machine-learning-2/classes/13138/), за любой дополнительной информацией - смотрите [документации HF](https://huggingface.co/docs).\n",
|
| 73 |
+
"\n",
|
| 74 |
+
"Начать лучше с простой модели, такой как [`distilbert-base-cased`](https://huggingface.co/distilbert-base-cased). Когда вы будете понимать, какие значения accuracy ожидать от базовой модели, можно поискать что-то получше. Два очевидных направления улучшения: (1) сильнее модель T5 или deberta v3, или (2) близкие данные, например взять модель которую предобучили на том же arxiv. И то и другое удобно [искать здесь](https://huggingface.co/models).\n",
|
| 75 |
+
"\n",
|
| 76 |
+
"## __Научили, и что теперь?__\n",
|
| 77 |
+
"\n",
|
| 78 |
+
"А теперь нужно сделать так, чтобы ваша обученная модель отвечала на запросы в интернете. Как и на прошлом этапе, вы можете сделать это кучей разных способов: от простого [streamlit](https://streamlit.io/) / [gradio](https://gradio.app/), минуя [TorchServe](https://pytorch.org/serve/) с [Triton/TensorRT](https://developer.nvidia.com/nvidia-triton-inference-server), и заканчивая экспортом модели в javascript с помощью [TensorFlow.js](https://www.tensorflow.org/js/tutorials) / [ONNX.js](https://github.com/elliotwaite/pytorch-to-javascript-with-onnx-js).\n",
|
| 79 |
+
"\n",
|
| 80 |
+
"На [семинаре](https://lk.yandexdataschool.ru/courses/2025-spring/7.1332-machine-learning-2/classes/13138/) мы разбирали основные вещи про то как работает streamlit и как сделать простое приложение с его помощью.\n",
|
| 81 |
+
"\n",
|
| 82 |
+
"Общая идея streamlit: вы [описываете](https://docs.streamlit.io/library/get-started/create-an-app) внешний вид приложения на питоне с помощью примитивов (кнопки, поля, любой html) -- а потом этот код выполняется на сервере и обслуживает каждого пользователя в отдельном процессе.\n",
|
| 83 |
+
"\n",
|
| 84 |
+
"__Для отладки__ можно запустить приложение локально, открыв консоль рядом с app.py:\n",
|
| 85 |
+
"* `pip install streamlit`\n",
|
| 86 |
+
"* `streamlit run app.py --server.port 8080`\n",
|
| 87 |
+
"* открыть в браузере localhost:8080, если он не открылся автоматически\n",
|
| 88 |
+
"\n",
|
| 89 |
+
"\n",
|
| 90 |
+
"## __Deployment time!__\n",
|
| 91 |
+
"\n",
|
| 92 |
+
"В этот раз вам нужно не просто написать код, __но и поднять ваше приложение с доступом из интернета__. И да, вы угадали, это можно сделать несколькими способами: [HuggingFace spaces](https://huggingface.co/spaces) (данный способ разбирали на [семинаре](https://lk.yandexdataschool.ru/courses/2025-spring/7.1332-machine-learning-2/classes/13138/)), [Streamlit Cloud](https://streamlit.io/cloud), а ещё вы можете купить или арендовать свой собственный сервер и захоститься там.\n",
|
| 93 |
+
"\n",
|
| 94 |
+
"Проще всего захостить на HF spaces, для этого вам нужно [зарегистрироваться](https://huggingface.co/join) и найти [меню создания нового приложения](https://huggingface.co/new-space). Название и лицензию можно выбрать на своё усмотрение, главное чтобы Space SDK был Streamlit, а доступ - public.\n",
|
| 95 |
+
"\n",
|
| 96 |
+
"Как создали - можно редактировать ваше приложение прямо на сайте, для этого откройте приложение и перейдите в Files and versions, и там в правом углу добавьте нужные файлы.\n",
|
| 97 |
+
"\n",
|
| 98 |
+
"На минималках вам потребуется 2 файла:\n",
|
| 99 |
+
"- `app.py`, о котором мы говорили выше\n",
|
| 100 |
+
"- `requirements.txt`, где вы укажете нужные вам библиотеки\n",
|
| 101 |
+
"\n",
|
| 102 |
+
"Вы можете разместить там же веса вашей обученной модели, любые необходимые данные, дополнительные файлы, ...\n",
|
| 103 |
+
"\n",
|
| 104 |
+
"После каждого изменения файлов, ваше приложение соберётся (обычно 1-5 минут) и будет доступно уже во вкладке App. Ну или не соберётся и покажет вам, где оно сломалось. И вуаля, теперь у вас есть ссылка, которую можно показать ~друзьям~ ассистентам курса и кому угодно в интернете.\n",
|
| 105 |
+
"\n",
|
| 106 |
+
"__Удобная работа с кодом.__ Пока у вас 2 файла, их легко редактивровать прямо в интерфейсе HF spaces. Если же у вас дюжина файлов, вам может быть удобнее редактировать их в любимом vscode/pycharm/.../emacs. Чтобы это не вызывало мучений, можно пользоваться HF spaces как git репозиторием ([подробности тут](https://huggingface.co/docs/hub/spaces#manage-app-with-github-actions)).\n",
|
| 107 |
+
"\n",
|
| 108 |
+
"## __Что нужно сдать__\n",
|
| 109 |
+
"\n",
|
| 110 |
+
"Вы сдаёте проект, который будет проверяться вручную, то что ожидается от каждого проекта:\n",
|
| 111 |
+
"- Текстовое сопровождение вашего конкретного проекта в любом удобно читаемом формате (pdf, html, текст в lk, ...) - что за задачу вы решали, где/как брали данные, какие использовали модели, какие проводили эксперименты, ...\n",
|
| 112 |
+
"- Ссылка на веб интерфейс, где можно протестировать демо вашего проекта - обязательно проверяйте что работает не только у вас (с другого устройства и из под incognito режима)\n",
|
| 113 |
+
"- Код обучения вашей модели (желательно ipynb с заполненными ячейками и не стёртыми выходами, переведённый в pdf / html), но если вы обучали не в ноутбуке, то сдавайте код в виде файла / архива файлов / git ссылки с readme.md описанием того как именно проходило обучение с помощью этого кода.\n",
|
| 114 |
+
"\n",
|
| 115 |
+
"## __Оценка__\n",
|
| 116 |
+
"\n",
|
| 117 |
+
"Мы будем оценивать проект целиком, включая идею и реализацию. Максимум за проект можно получить 10 баллов, но мы оставляем ещё до 5 баллов, котор��е можем выдать как бонусные за особенно интересные и качественно реализованные проекты.\n",
|
| 118 |
+
"\n",
|
| 119 |
+
"### __Тонкие места, за которые могут быть снижения баллов:__\n",
|
| 120 |
+
"\n",
|
| 121 |
+
"__1. Скорость работы.__\n",
|
| 122 |
+
"\n",
|
| 123 |
+
"По умолчанию, streamlit будет выполняет весь ваш код на каждое действие пользователя. То есть всякий раз, когда пользователь меняет что-то в тексте, оно будет заново загружать модель. Чтобы исправить это безобразие, вы можете закэшировать подготовленную модель в `@st.cache`. Подробности в [семинаре](https://lk.yandexdataschool.ru/courses/2025-spring/7.1332-machine-learning-2/classes/13138/), а также [читайте тут](https://docs.streamlit.io/library/advanced-features/caching).\n",
|
| 124 |
+
"\n",
|
| 125 |
+
"__Как будет оцениваться:__\n",
|
| 126 |
+
"\n",
|
| 127 |
+
"Вы не обязаны пользоваться кэшированием, но ваше приложение не должно неоправдано тормозить дольше, чем на 3 секунды. \"Оправданые\" тормоза это те, которые вы явно оправдали текстом в ЛМС :)\n",
|
| 128 |
+
"\n",
|
| 129 |
+
"-----\n",
|
| 130 |
+
"\n",
|
| 131 |
+
"__2. Понятный фронтенд.__\n",
|
| 132 |
+
"\n",
|
| 133 |
+
"Наколеночный графический интерфейс с семинара - пример того, как скорее не надо делать интерфейс приложения. Как надо - сложный вопрос, причём настолько сложный, что есть даже [Школа Разработки Интерфейсов](https://academy.yandex.ru/schools/frontend). Но для начала:\n",
|
| 134 |
+
"\n",
|
| 135 |
+
"- Выводить нужно человекочитаемый текст, а не просто JSON с индексами и метаданными.\n",
|
| 136 |
+
"- Пользователю должно быть понятно, куда и какие данные вводить. Пустые текстовые поля в вакууме - плохой тон.\n",
|
| 137 |
+
"- Сервис не должен падать с не_отловленными ошибками. Даже если пользователь введёт неправильные/пустые данные, нужно это обработать и написать, где произошла ошибка.\n",
|
| 138 |
+
"\n",
|
| 139 |
+
"__Как будет оцениваться:__\n",
|
| 140 |
+
"\n",
|
| 141 |
+
"Для полного балла достаточно соблюсти эти три правила и специально не стрелять себе в ногу.\n",
|
| 142 |
+
"\n",
|
| 143 |
+
"-----\n",
|
| 144 |
+
"\n",
|
| 145 |
+
"__3. Код обучения и инференса.__\n",
|
| 146 |
+
"\n",
|
| 147 |
+
"Сдавая проект мы будем также получать от вас код проекта (как обучения ваших моделей, так и код веб интерфейса).\n",
|
| 148 |
+
"\n",
|
| 149 |
+
"__Как будет оцениваться:__\n",
|
| 150 |
+
"\n",
|
| 151 |
+
"Код не будет отдельно проверяться как часть задания, поэтому пишите как хотите, однако - в спорных ситуациях мы оставляем за собой право проверить ваш код, за чем могут последовать потенциальные снижения баллов при любых нарушениях.\n"
|
| 152 |
+
]
|
| 153 |
+
},
|
| 154 |
+
{
|
| 155 |
+
"cell_type": "markdown",
|
| 156 |
+
"metadata": {},
|
| 157 |
+
"source": []
|
| 158 |
+
}
|
| 159 |
+
],
|
| 160 |
+
"metadata": {
|
| 161 |
+
"kernelspec": {
|
| 162 |
+
"display_name": "Python 3 (ipykernel)",
|
| 163 |
+
"language": "python",
|
| 164 |
+
"name": "python3"
|
| 165 |
+
},
|
| 166 |
+
"language_info": {
|
| 167 |
+
"codemirror_mode": {
|
| 168 |
+
"name": "ipython",
|
| 169 |
+
"version": 3
|
| 170 |
+
},
|
| 171 |
+
"file_extension": ".py",
|
| 172 |
+
"mimetype": "text/x-python",
|
| 173 |
+
"name": "python",
|
| 174 |
+
"nbconvert_exporter": "python",
|
| 175 |
+
"pygments_lexer": "ipython3",
|
| 176 |
+
"version": "3.10.12"
|
| 177 |
+
},
|
| 178 |
+
"widgets": {
|
| 179 |
+
"application/vnd.jupyter.widget-state+json": {
|
| 180 |
+
"state": {},
|
| 181 |
+
"version_major": 2,
|
| 182 |
+
"version_minor": 0
|
| 183 |
+
}
|
| 184 |
+
}
|
| 185 |
+
},
|
| 186 |
+
"nbformat": 4,
|
| 187 |
+
"nbformat_minor": 4
|
| 188 |
+
}
|
app.py
ADDED
|
@@ -0,0 +1,73 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import streamlit as st
|
| 2 |
+
|
| 3 |
+
from transformers import AutoTokenizer, DistilBertForSequenceClassification
|
| 4 |
+
import torch
|
| 5 |
+
from torch.nn.functional import softmax
|
| 6 |
+
|
| 7 |
+
|
| 8 |
+
|
| 9 |
+
base_model_name = 'distilbert-base-uncased'
|
| 10 |
+
|
| 11 |
+
@st.cache
|
| 12 |
+
def load_tags_info():
|
| 13 |
+
|
| 14 |
+
id_to_description = {}
|
| 15 |
+
with open('tags.txt', 'r') as file:
|
| 16 |
+
i = 0
|
| 17 |
+
for line in file:
|
| 18 |
+
|
| 19 |
+
description = line[:-1]
|
| 20 |
+
|
| 21 |
+
id_to_description[i] = description
|
| 22 |
+
|
| 23 |
+
i += 1
|
| 24 |
+
|
| 25 |
+
return id_to_description
|
| 26 |
+
|
| 27 |
+
id_to_description = load_tags_info()
|
| 28 |
+
|
| 29 |
+
@st.cache
|
| 30 |
+
def load_model():
|
| 31 |
+
return DistilBertForSequenceClassification.from_pretrained('./')
|
| 32 |
+
|
| 33 |
+
def load_tokenizer():
|
| 34 |
+
return AutoTokenizer.from_pretrained('distilbert-base-uncased')
|
| 35 |
+
|
| 36 |
+
def top_xx(preds, xx=95):
|
| 37 |
+
tops = torch.argsort(preds, 1, descending=True)
|
| 38 |
+
total = 0
|
| 39 |
+
index = 0
|
| 40 |
+
result = []
|
| 41 |
+
while total < xx / 100:
|
| 42 |
+
next_id = tops[0, index].item()
|
| 43 |
+
total += preds[0, next_id]
|
| 44 |
+
index += 1
|
| 45 |
+
result.append(id_to_description[next_id])
|
| 46 |
+
return result
|
| 47 |
+
|
| 48 |
+
model = load_model()
|
| 49 |
+
tokenizer = load_tokenizer()
|
| 50 |
+
temperature = 1
|
| 51 |
+
|
| 52 |
+
st.title('ArXivTager')
|
| 53 |
+
st.caption('Напишите тему (Title) и параграф из статьи (Abstract). Поля должны быть ЗАПОЛНЕНЫ для корректной классификации.')
|
| 54 |
+
|
| 55 |
+
with st.form("ArXivTager"):
|
| 56 |
+
|
| 57 |
+
title = st.text_area(label='Title', height=30)
|
| 58 |
+
abstract = st.text_area(label='Abstract (optional)', height=200)
|
| 59 |
+
st.caption('ВЫВОД: набор тем в порядке уменьшения вероятностей.')
|
| 60 |
+
|
| 61 |
+
submitted = st.form_submit_button("Get tags")
|
| 62 |
+
if submitted:
|
| 63 |
+
if title == '':
|
| 64 |
+
st.markdown("Нужно хоть что-то написать")
|
| 65 |
+
else:
|
| 66 |
+
prompt = 'Title: ' + title + ' Abstract: ' + abstract
|
| 67 |
+
tokens = tokenizer(prompt, truncation=True, padding='max_length', return_tensors='pt')['input_ids']
|
| 68 |
+
preds = softmax(model(tokens.reshape(1, -1)).logits / temperature, dim=1)
|
| 69 |
+
tags = top_xx(preds)
|
| 70 |
+
other_tags = []
|
| 71 |
+
st.header('Inferred tags:')
|
| 72 |
+
for i, tag_data in enumerate(tags):
|
| 73 |
+
st.markdown('* ' + tag_data)
|
config.json
ADDED
|
@@ -0,0 +1,338 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_name_or_path": "distilbert-base-uncased",
|
| 3 |
+
"activation": "gelu",
|
| 4 |
+
"architectures": [
|
| 5 |
+
"DistilBertForSequenceClassification"
|
| 6 |
+
],
|
| 7 |
+
"attention_dropout": 0.1,
|
| 8 |
+
"dim": 768,
|
| 9 |
+
"dropout": 0.1,
|
| 10 |
+
"hidden_dim": 3072,
|
| 11 |
+
"id2label": {
|
| 12 |
+
"0": "LABEL_0",
|
| 13 |
+
"1": "LABEL_1",
|
| 14 |
+
"2": "LABEL_2",
|
| 15 |
+
"3": "LABEL_3",
|
| 16 |
+
"4": "LABEL_4",
|
| 17 |
+
"5": "LABEL_5",
|
| 18 |
+
"6": "LABEL_6",
|
| 19 |
+
"7": "LABEL_7",
|
| 20 |
+
"8": "LABEL_8",
|
| 21 |
+
"9": "LABEL_9",
|
| 22 |
+
"10": "LABEL_10",
|
| 23 |
+
"11": "LABEL_11",
|
| 24 |
+
"12": "LABEL_12",
|
| 25 |
+
"13": "LABEL_13",
|
| 26 |
+
"14": "LABEL_14",
|
| 27 |
+
"15": "LABEL_15",
|
| 28 |
+
"16": "LABEL_16",
|
| 29 |
+
"17": "LABEL_17",
|
| 30 |
+
"18": "LABEL_18",
|
| 31 |
+
"19": "LABEL_19",
|
| 32 |
+
"20": "LABEL_20",
|
| 33 |
+
"21": "LABEL_21",
|
| 34 |
+
"22": "LABEL_22",
|
| 35 |
+
"23": "LABEL_23",
|
| 36 |
+
"24": "LABEL_24",
|
| 37 |
+
"25": "LABEL_25",
|
| 38 |
+
"26": "LABEL_26",
|
| 39 |
+
"27": "LABEL_27",
|
| 40 |
+
"28": "LABEL_28",
|
| 41 |
+
"29": "LABEL_29",
|
| 42 |
+
"30": "LABEL_30",
|
| 43 |
+
"31": "LABEL_31",
|
| 44 |
+
"32": "LABEL_32",
|
| 45 |
+
"33": "LABEL_33",
|
| 46 |
+
"34": "LABEL_34",
|
| 47 |
+
"35": "LABEL_35",
|
| 48 |
+
"36": "LABEL_36",
|
| 49 |
+
"37": "LABEL_37",
|
| 50 |
+
"38": "LABEL_38",
|
| 51 |
+
"39": "LABEL_39",
|
| 52 |
+
"40": "LABEL_40",
|
| 53 |
+
"41": "LABEL_41",
|
| 54 |
+
"42": "LABEL_42",
|
| 55 |
+
"43": "LABEL_43",
|
| 56 |
+
"44": "LABEL_44",
|
| 57 |
+
"45": "LABEL_45",
|
| 58 |
+
"46": "LABEL_46",
|
| 59 |
+
"47": "LABEL_47",
|
| 60 |
+
"48": "LABEL_48",
|
| 61 |
+
"49": "LABEL_49",
|
| 62 |
+
"50": "LABEL_50",
|
| 63 |
+
"51": "LABEL_51",
|
| 64 |
+
"52": "LABEL_52",
|
| 65 |
+
"53": "LABEL_53",
|
| 66 |
+
"54": "LABEL_54",
|
| 67 |
+
"55": "LABEL_55",
|
| 68 |
+
"56": "LABEL_56",
|
| 69 |
+
"57": "LABEL_57",
|
| 70 |
+
"58": "LABEL_58",
|
| 71 |
+
"59": "LABEL_59",
|
| 72 |
+
"60": "LABEL_60",
|
| 73 |
+
"61": "LABEL_61",
|
| 74 |
+
"62": "LABEL_62",
|
| 75 |
+
"63": "LABEL_63",
|
| 76 |
+
"64": "LABEL_64",
|
| 77 |
+
"65": "LABEL_65",
|
| 78 |
+
"66": "LABEL_66",
|
| 79 |
+
"67": "LABEL_67",
|
| 80 |
+
"68": "LABEL_68",
|
| 81 |
+
"69": "LABEL_69",
|
| 82 |
+
"70": "LABEL_70",
|
| 83 |
+
"71": "LABEL_71",
|
| 84 |
+
"72": "LABEL_72",
|
| 85 |
+
"73": "LABEL_73",
|
| 86 |
+
"74": "LABEL_74",
|
| 87 |
+
"75": "LABEL_75",
|
| 88 |
+
"76": "LABEL_76",
|
| 89 |
+
"77": "LABEL_77",
|
| 90 |
+
"78": "LABEL_78",
|
| 91 |
+
"79": "LABEL_79",
|
| 92 |
+
"80": "LABEL_80",
|
| 93 |
+
"81": "LABEL_81",
|
| 94 |
+
"82": "LABEL_82",
|
| 95 |
+
"83": "LABEL_83",
|
| 96 |
+
"84": "LABEL_84",
|
| 97 |
+
"85": "LABEL_85",
|
| 98 |
+
"86": "LABEL_86",
|
| 99 |
+
"87": "LABEL_87",
|
| 100 |
+
"88": "LABEL_88",
|
| 101 |
+
"89": "LABEL_89",
|
| 102 |
+
"90": "LABEL_90",
|
| 103 |
+
"91": "LABEL_91",
|
| 104 |
+
"92": "LABEL_92",
|
| 105 |
+
"93": "LABEL_93",
|
| 106 |
+
"94": "LABEL_94",
|
| 107 |
+
"95": "LABEL_95",
|
| 108 |
+
"96": "LABEL_96",
|
| 109 |
+
"97": "LABEL_97",
|
| 110 |
+
"98": "LABEL_98",
|
| 111 |
+
"99": "LABEL_99",
|
| 112 |
+
"100": "LABEL_100",
|
| 113 |
+
"101": "LABEL_101",
|
| 114 |
+
"102": "LABEL_102",
|
| 115 |
+
"103": "LABEL_103",
|
| 116 |
+
"104": "LABEL_104",
|
| 117 |
+
"105": "LABEL_105",
|
| 118 |
+
"106": "LABEL_106",
|
| 119 |
+
"107": "LABEL_107",
|
| 120 |
+
"108": "LABEL_108",
|
| 121 |
+
"109": "LABEL_109",
|
| 122 |
+
"110": "LABEL_110",
|
| 123 |
+
"111": "LABEL_111",
|
| 124 |
+
"112": "LABEL_112",
|
| 125 |
+
"113": "LABEL_113",
|
| 126 |
+
"114": "LABEL_114",
|
| 127 |
+
"115": "LABEL_115",
|
| 128 |
+
"116": "LABEL_116",
|
| 129 |
+
"117": "LABEL_117",
|
| 130 |
+
"118": "LABEL_118",
|
| 131 |
+
"119": "LABEL_119",
|
| 132 |
+
"120": "LABEL_120",
|
| 133 |
+
"121": "LABEL_121",
|
| 134 |
+
"122": "LABEL_122",
|
| 135 |
+
"123": "LABEL_123",
|
| 136 |
+
"124": "LABEL_124",
|
| 137 |
+
"125": "LABEL_125",
|
| 138 |
+
"126": "LABEL_126",
|
| 139 |
+
"127": "LABEL_127",
|
| 140 |
+
"128": "LABEL_128",
|
| 141 |
+
"129": "LABEL_129",
|
| 142 |
+
"130": "LABEL_130",
|
| 143 |
+
"131": "LABEL_131",
|
| 144 |
+
"132": "LABEL_132",
|
| 145 |
+
"133": "LABEL_133",
|
| 146 |
+
"134": "LABEL_134",
|
| 147 |
+
"135": "LABEL_135",
|
| 148 |
+
"136": "LABEL_136",
|
| 149 |
+
"137": "LABEL_137",
|
| 150 |
+
"138": "LABEL_138",
|
| 151 |
+
"139": "LABEL_139",
|
| 152 |
+
"140": "LABEL_140",
|
| 153 |
+
"141": "LABEL_141",
|
| 154 |
+
"142": "LABEL_142",
|
| 155 |
+
"143": "LABEL_143",
|
| 156 |
+
"144": "LABEL_144",
|
| 157 |
+
"145": "LABEL_145",
|
| 158 |
+
"146": "LABEL_146",
|
| 159 |
+
"147": "LABEL_147",
|
| 160 |
+
"148": "LABEL_148",
|
| 161 |
+
"149": "LABEL_149",
|
| 162 |
+
"150": "LABEL_150",
|
| 163 |
+
"151": "LABEL_151",
|
| 164 |
+
"152": "LABEL_152",
|
| 165 |
+
"153": "LABEL_153",
|
| 166 |
+
"154": "LABEL_154"
|
| 167 |
+
},
|
| 168 |
+
"initializer_range": 0.02,
|
| 169 |
+
"label2id": {
|
| 170 |
+
"LABEL_0": 0,
|
| 171 |
+
"LABEL_1": 1,
|
| 172 |
+
"LABEL_10": 10,
|
| 173 |
+
"LABEL_100": 100,
|
| 174 |
+
"LABEL_101": 101,
|
| 175 |
+
"LABEL_102": 102,
|
| 176 |
+
"LABEL_103": 103,
|
| 177 |
+
"LABEL_104": 104,
|
| 178 |
+
"LABEL_105": 105,
|
| 179 |
+
"LABEL_106": 106,
|
| 180 |
+
"LABEL_107": 107,
|
| 181 |
+
"LABEL_108": 108,
|
| 182 |
+
"LABEL_109": 109,
|
| 183 |
+
"LABEL_11": 11,
|
| 184 |
+
"LABEL_110": 110,
|
| 185 |
+
"LABEL_111": 111,
|
| 186 |
+
"LABEL_112": 112,
|
| 187 |
+
"LABEL_113": 113,
|
| 188 |
+
"LABEL_114": 114,
|
| 189 |
+
"LABEL_115": 115,
|
| 190 |
+
"LABEL_116": 116,
|
| 191 |
+
"LABEL_117": 117,
|
| 192 |
+
"LABEL_118": 118,
|
| 193 |
+
"LABEL_119": 119,
|
| 194 |
+
"LABEL_12": 12,
|
| 195 |
+
"LABEL_120": 120,
|
| 196 |
+
"LABEL_121": 121,
|
| 197 |
+
"LABEL_122": 122,
|
| 198 |
+
"LABEL_123": 123,
|
| 199 |
+
"LABEL_124": 124,
|
| 200 |
+
"LABEL_125": 125,
|
| 201 |
+
"LABEL_126": 126,
|
| 202 |
+
"LABEL_127": 127,
|
| 203 |
+
"LABEL_128": 128,
|
| 204 |
+
"LABEL_129": 129,
|
| 205 |
+
"LABEL_13": 13,
|
| 206 |
+
"LABEL_130": 130,
|
| 207 |
+
"LABEL_131": 131,
|
| 208 |
+
"LABEL_132": 132,
|
| 209 |
+
"LABEL_133": 133,
|
| 210 |
+
"LABEL_134": 134,
|
| 211 |
+
"LABEL_135": 135,
|
| 212 |
+
"LABEL_136": 136,
|
| 213 |
+
"LABEL_137": 137,
|
| 214 |
+
"LABEL_138": 138,
|
| 215 |
+
"LABEL_139": 139,
|
| 216 |
+
"LABEL_14": 14,
|
| 217 |
+
"LABEL_140": 140,
|
| 218 |
+
"LABEL_141": 141,
|
| 219 |
+
"LABEL_142": 142,
|
| 220 |
+
"LABEL_143": 143,
|
| 221 |
+
"LABEL_144": 144,
|
| 222 |
+
"LABEL_145": 145,
|
| 223 |
+
"LABEL_146": 146,
|
| 224 |
+
"LABEL_147": 147,
|
| 225 |
+
"LABEL_148": 148,
|
| 226 |
+
"LABEL_149": 149,
|
| 227 |
+
"LABEL_15": 15,
|
| 228 |
+
"LABEL_150": 150,
|
| 229 |
+
"LABEL_151": 151,
|
| 230 |
+
"LABEL_152": 152,
|
| 231 |
+
"LABEL_153": 153,
|
| 232 |
+
"LABEL_154": 154,
|
| 233 |
+
"LABEL_16": 16,
|
| 234 |
+
"LABEL_17": 17,
|
| 235 |
+
"LABEL_18": 18,
|
| 236 |
+
"LABEL_19": 19,
|
| 237 |
+
"LABEL_2": 2,
|
| 238 |
+
"LABEL_20": 20,
|
| 239 |
+
"LABEL_21": 21,
|
| 240 |
+
"LABEL_22": 22,
|
| 241 |
+
"LABEL_23": 23,
|
| 242 |
+
"LABEL_24": 24,
|
| 243 |
+
"LABEL_25": 25,
|
| 244 |
+
"LABEL_26": 26,
|
| 245 |
+
"LABEL_27": 27,
|
| 246 |
+
"LABEL_28": 28,
|
| 247 |
+
"LABEL_29": 29,
|
| 248 |
+
"LABEL_3": 3,
|
| 249 |
+
"LABEL_30": 30,
|
| 250 |
+
"LABEL_31": 31,
|
| 251 |
+
"LABEL_32": 32,
|
| 252 |
+
"LABEL_33": 33,
|
| 253 |
+
"LABEL_34": 34,
|
| 254 |
+
"LABEL_35": 35,
|
| 255 |
+
"LABEL_36": 36,
|
| 256 |
+
"LABEL_37": 37,
|
| 257 |
+
"LABEL_38": 38,
|
| 258 |
+
"LABEL_39": 39,
|
| 259 |
+
"LABEL_4": 4,
|
| 260 |
+
"LABEL_40": 40,
|
| 261 |
+
"LABEL_41": 41,
|
| 262 |
+
"LABEL_42": 42,
|
| 263 |
+
"LABEL_43": 43,
|
| 264 |
+
"LABEL_44": 44,
|
| 265 |
+
"LABEL_45": 45,
|
| 266 |
+
"LABEL_46": 46,
|
| 267 |
+
"LABEL_47": 47,
|
| 268 |
+
"LABEL_48": 48,
|
| 269 |
+
"LABEL_49": 49,
|
| 270 |
+
"LABEL_5": 5,
|
| 271 |
+
"LABEL_50": 50,
|
| 272 |
+
"LABEL_51": 51,
|
| 273 |
+
"LABEL_52": 52,
|
| 274 |
+
"LABEL_53": 53,
|
| 275 |
+
"LABEL_54": 54,
|
| 276 |
+
"LABEL_55": 55,
|
| 277 |
+
"LABEL_56": 56,
|
| 278 |
+
"LABEL_57": 57,
|
| 279 |
+
"LABEL_58": 58,
|
| 280 |
+
"LABEL_59": 59,
|
| 281 |
+
"LABEL_6": 6,
|
| 282 |
+
"LABEL_60": 60,
|
| 283 |
+
"LABEL_61": 61,
|
| 284 |
+
"LABEL_62": 62,
|
| 285 |
+
"LABEL_63": 63,
|
| 286 |
+
"LABEL_64": 64,
|
| 287 |
+
"LABEL_65": 65,
|
| 288 |
+
"LABEL_66": 66,
|
| 289 |
+
"LABEL_67": 67,
|
| 290 |
+
"LABEL_68": 68,
|
| 291 |
+
"LABEL_69": 69,
|
| 292 |
+
"LABEL_7": 7,
|
| 293 |
+
"LABEL_70": 70,
|
| 294 |
+
"LABEL_71": 71,
|
| 295 |
+
"LABEL_72": 72,
|
| 296 |
+
"LABEL_73": 73,
|
| 297 |
+
"LABEL_74": 74,
|
| 298 |
+
"LABEL_75": 75,
|
| 299 |
+
"LABEL_76": 76,
|
| 300 |
+
"LABEL_77": 77,
|
| 301 |
+
"LABEL_78": 78,
|
| 302 |
+
"LABEL_79": 79,
|
| 303 |
+
"LABEL_8": 8,
|
| 304 |
+
"LABEL_80": 80,
|
| 305 |
+
"LABEL_81": 81,
|
| 306 |
+
"LABEL_82": 82,
|
| 307 |
+
"LABEL_83": 83,
|
| 308 |
+
"LABEL_84": 84,
|
| 309 |
+
"LABEL_85": 85,
|
| 310 |
+
"LABEL_86": 86,
|
| 311 |
+
"LABEL_87": 87,
|
| 312 |
+
"LABEL_88": 88,
|
| 313 |
+
"LABEL_89": 89,
|
| 314 |
+
"LABEL_9": 9,
|
| 315 |
+
"LABEL_90": 90,
|
| 316 |
+
"LABEL_91": 91,
|
| 317 |
+
"LABEL_92": 92,
|
| 318 |
+
"LABEL_93": 93,
|
| 319 |
+
"LABEL_94": 94,
|
| 320 |
+
"LABEL_95": 95,
|
| 321 |
+
"LABEL_96": 96,
|
| 322 |
+
"LABEL_97": 97,
|
| 323 |
+
"LABEL_98": 98,
|
| 324 |
+
"LABEL_99": 99
|
| 325 |
+
},
|
| 326 |
+
"max_position_embeddings": 512,
|
| 327 |
+
"model_type": "distilbert",
|
| 328 |
+
"n_heads": 12,
|
| 329 |
+
"n_layers": 6,
|
| 330 |
+
"pad_token_id": 0,
|
| 331 |
+
"qa_dropout": 0.1,
|
| 332 |
+
"seq_classif_dropout": 0.2,
|
| 333 |
+
"sinusoidal_pos_embds": false,
|
| 334 |
+
"tie_weights_": true,
|
| 335 |
+
"torch_dtype": "float32",
|
| 336 |
+
"transformers_version": "4.28.1",
|
| 337 |
+
"vocab_size": 30522
|
| 338 |
+
}
|
model_train.ipynb
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5c8f30540e3f618551d9d8a2c2f8a76cfcf9008187df7526eeab7a6e4d731c9d
|
| 3 |
+
size 268326125
|
requirements.txt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
numpy
|
| 2 |
+
torch
|
| 3 |
+
transformers
|