Go to file
2023-05-10 22:44:14 +02:00
teng-ml Fixed model, restructured files 2023-05-10 22:44:14 +02:00
.gitignore Fixed model, restructured files 2023-05-10 22:44:14 +02:00
readme.md Fixed model, restructured files 2023-05-10 22:44:14 +02:00

Machine Learning stuff for TENG project

(Bi)LSTM for name classification.
More information on the project are on my website.

Model training

Adjust the parameters in main.py and run it.
All models and the settings they were trained with are automatically serialized with pickle and stored in a subfolder of the <model_dir> that was set in main.py.

Model evaluation

Run find_best_model.py <model_dir> with the <model_dir> specified in main.py during training.