Skip to content

Simple python project that evaluates multiple ML algorithms in order to get know which algorithm performs the best

Notifications You must be signed in to change notification settings

FilipeGood/Evaluate-ML-Algorithms

Repository files navigation

Evaluate-ML-Algorithms

Simple python project that evaluates multiple ML algorithms in order to get know which algorithm performs the best

  • This simple project, takes the data and fits iteratively multiple machine learning algorithms using sklearn cross_validate function.

Usage:

python3 evaluate_multiple_algorithms.py -d <dataset_path> -p <classification or regression> -t <target column name>

After fitting, the program reports the results.

First it shows the results dataframe:

Screenshot

And then, it builds two plots:

Bar plot with the fit times

Screenshot

Bar plot with the mean of a certain metric (for example accuracy)

Screenshot

About

Simple python project that evaluates multiple ML algorithms in order to get know which algorithm performs the best

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages