mosievich.kirill commited on
Commit
2c26172
·
1 Parent(s): 779c97b

task_4 baseline

Browse files
Files changed (6) hide show
  1. ML2_2025_nlp_ops1.ipynb +188 -0
  2. app.py +73 -0
  3. config.json +338 -0
  4. model_train.ipynb +0 -0
  5. pytorch_model.bin +3 -0
  6. 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
+ "![img](https://d35w6hwqhdq0in.cloudfront.net/521712556725591dcacec5bbdb32e047.png)\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