A demo to show scikit-learn to implement regression and classification.
The repo contains the following files:
regression.py: a simple Linear Regression model to predict the Salary of an employee based on the years of experience. It works with the 'Salary_Data.csv' file.Salary_Data.csv: the dataset used in the regression.py script.classification.py: a simple Random Forest classification model to classify into one of three flower species from the Iris public dataset.