Projekt na przedmiot Uczenie Maszynowe, semestr letni 2021/2022.
bert_model | ||
lstm_model | ||
raport.docx | ||
raport.pdf | ||
README.md | ||
requirements.txt | ||
run_models.py | ||
stopwords_filter.py | ||
test.txt | ||
train_bert.py | ||
train_lstm.py | ||
train_naive_bayes.py | ||
train.txt |
- Do obejrzenia wszystkiego zrealizowanego w ramach projektu powinno wystarczyć uruchomienie skryptu run_models.py i podążanie zgodnie z wyświetlanymi instrukcjami.
- Poprzez kilkusekundowe czasy ładowania różnych komponentów run_models.py zdaje się czasami nie reagować na wciskane klawisze, wtedy najprościej wcisnąć jakiś kilka razy pod rząd.
- Modele LSTM i BERT łącznie trenują się dosyć długo, szczególnie w przypadku braku dedykowanego GPU, dlatego już wytrenowane zostały załączone w odpowiednio nazwanych katalogach. Projekt skorzysta z nich, jeśli będą dostępne.