ttarnawski / open_x_task

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

open_x_task

'data' folder contains data for classification problem.
'ml_models.py' contains required ML models: heuristic (kNN), 2 baseline models (Decision Tree, Logistic Regression) and a neural network model.
'rest_api.py' contains REST API in Flask, user chooses one of four models, inputs parameters and gets accuracy score, confusion matrix and downloadable csv file with predicted labels.
'templates' folder contains all necessary html templates for REST API.
'models_evaluations.py' contains a one class that defines a method for each model to plot accuracy metric based on given range of parameters for three models, and one method for neural network to evaluate and pick best hyperparameters.
'requiremets.txt' contains required libraries to run the REST API.

About


Languages

Language:Python 79.4%Language:HTML 20.6%