MatthiasQuintern/teng-ml
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
22 Commits 2 Branches 0 Tags 754 KiB
Python 100%
ad2e3468f7
Go to file
Matthias@Dell ad2e3468f7 add n_object to regex
2023-08-04 13:37:45 +02:00
teng_ml add n_object to regex 2023-08-04 13:37:45 +02:00
.gitignore Fixed model, restructured files 2023-05-10 22:44:14 +02:00
LICENSE Made installable 2023-05-26 14:01:15 +02:00
pyproject.toml Made installable 2023-05-26 14:01:15 +02:00
readme.md Better title 2023-05-10 22:46:23 +02:00

readme.md

Machine learning for material recognition with a TENG

(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.