not really known
Go to file
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

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

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