jarvis.ai.pkgs.sklearn.regression

Simple ML models for regression.

Designed for educational purposes only.

Module Contents

Functions

regression([X, Y, plot, models, preprocess, test_size])

Provide model as models to get accuracy.

Attributes

simple_regr_models

jarvis.ai.pkgs.sklearn.regression.simple_regr_models
jarvis.ai.pkgs.sklearn.regression.regression(X=[], Y=[], plot=False, models=simple_regr_models, preprocess=True, test_size=0.1)[source]

Provide model as models to get accuracy.

Args:

X: input features

Y: Target data

models : collection array of models

plot: whether to make a parity plot with ML models

preprocess: whether to apply standard preprocessing techniques