mirror of
https://github.com/magnum-opus-nn-cp/ml.git
synced 2024-11-10 18:26:35 +03:00
11 lines
807 B
Markdown
11 lines
807 B
Markdown
|
### Установить зависимости
|
|||
|
|
`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
|