mlinsights extends scikit-learn with a couple of new models, transformers, metrics, plotting. It provides new trainers such as QuantileLinearRegression which trains a linear regression with L1 norm non-linear correlation based on decision trees, or QuantileMLPRegressor a modification of scikit-learn's MLPRegressor which trains a multi-layer perceptron with L1 norm. It also explores PredictableTSNE which trains a supervized model to replicate t-SNE results or a PiecewiseRegression which partitions the data before fitting a model on each bucket.
yuyangl / mlinsights Goto Github PK
View Code? Open in Web Editor NEWThis project forked from sdpython/mlinsights
mlinsights extends scikit-learn with a couple of new models, transformers, metrics, plotting.
Home Page: http://www.xavierdupre.fr/app/mlinsights/helpsphinx/index.html
License: MIT License