magnum-opus-nn-cp/ml
1
0
Fork 0
mirror of https://github.com/magnum-opus-nn-cp/ml.git synced 2024-11-21 23:46:33 +03:00
Code Issues Packages Projects Releases Wiki Activity
not really known
1 Commit 1 Branch 0 Tags 54 KiB
Jupyter Notebook 73.5%
Python 26.5%
e548bd4ad7
Go to file
илья e548bd4ad7 upload ml
2023-09-10 08:43:58 +03:00
.DS_Store upload ml 2023-09-10 08:43:58 +03:00
annoy_labels.pickle upload ml 2023-09-10 08:43:58 +03:00
bert_train.ipynb upload ml 2023-09-10 08:43:58 +03:00
catboost-train.ipynb upload ml 2023-09-10 08:43:58 +03:00
inference.py upload ml 2023-09-10 08:43:58 +03:00
nearest-search-train.ipynb upload ml 2023-09-10 08:43:58 +03:00
README.md upload ml 2023-09-10 08:43:58 +03:00
req.txt upload ml 2023-09-10 08:43:58 +03:00
tfidf-train.ipynb upload ml 2023-09-10 08:43:58 +03:00

README.md

Установить зависимости

pip3 install -r req.txt

Локации документов для обучения

  • catboost-train.ipynb - файл для обучения кэтбуста
  • bert_train.ipynb - файл для обучения берта
  • nearest-search-train.ipynb - файл для обучения поиска близжайших соседей
  • tfidf-train.ipynb - файл для обучения tf-idf + random forest

Инференс модели

uvicorn inference:app --reload --workers 1

У мля слишком много больших файлов поэтому мы выложили код с весами моделей на гугл диск https://drive.google.com/drive/folders/1hnWKpZjtQLBbzAE9YsUW_4x-IEb3mFvg?usp=sharing