Classification train first models - Githubissues

maxivmn / pklot

MIT License

1 stars 2 forks source link

Classification train first models #8

Open RitkaMargaritka opened 10 months ago

RitkaMargaritka commented 10 months ago

Choose baseline model
train 3 models in total
track properly in MLFlow.
Evaluate the models.

RitkaMargaritka commented 10 months ago

Baselinemodel: SVM Other classification models:

ResNet18
MobileNetV3_small
very small model