automl_tools
Project description
Automl_tools: automl binary classification
Automl_tools is a Python library that implements Gradient Boosting
Installation
The code is packaged for PyPI, so that the installation consists in running:
pip install automl-tools
Colab
Usage
Probabilistic binary example on the Boston housing dataset:
import pandas as pd
from automl_tools import automl_run
train = pd.read_csv("https://raw.githubusercontent.com/jonaqp/automl_tools/main/automl_tools/examples/train.csv?token=AAN2ZBDWF77QITK4ARSFIFDABUGAU")
test = pd.read_csv("https://raw.githubusercontent.com/jonaqp/automl_tools/main/automl_tools/examples/test.csv?token=AAN2ZBD6TMUC5XSGRTJNVPDABUGCO")
automl_run(train=train,
test=test,
target_col="Survived",
imp_num="knn",
imp_cat="knn",
processing="binding",
mutual_information=False,
correlation_drop=False,
model_feature_selection=None,
model_run="LR",
augmentation=True,
Stratified=True)
Parameter
imp_num : "gaussian", "arbitrary", "median", "mean", "random", "knn"
imp_cat : "frequent", "constant", "rare", "knn"
processing: "woe", "binding"
Support Binary
model_feature_selection:
default: ["LR", "RF", "LGB"]
LR : LogisticRegression
RF : RandomForestClassifier
SVM : SVC
LS : LASSO
RD : RIDGE
NET : Elasticnet
DT : DecisionTreeClassifier
ET : ExtraTreesClassifier
GB : GradientBoostingClassifier
AB : AdaBoostClassifier
XGB : XGBClassifier
LGB : LGBMClassifier
CTB : CatBoostClassifier
NGB : NGBClassifier
model_run:
default: "LR"
LR : LogisticRegression
RF : RandomForestClassifier
SVM : SVC
LS : LASSO
RD : RIDGE
NET : Elasticnet
DT : DecisionTreeClassifier
ET : ExtraTreesClassifier
GB : GradientBoostingClassifier
AB : AdaBoostClassifier
XGB : XGBClassifier
LGB : LGBMClassifier
CTB : CatBoostClassifier
NGB : NGBClassifier
License
New features v1.0
- multi_class
- regression
- integrations GCP deploy model CI/CD
- integrations AWS deploy model CI/CD
BugFix
- 0.1.2
- add parameter id_col
- add comments readme.txt
Reference
- Jonathan Quiza github.
- Jonathan Quiza RumiMLSpark.
- Jonathan Quiza linkedin.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
automl_tools-0.1.2.tar.gz
(17.1 kB
view hashes)
Built Distribution
Close
Hashes for automl_tools-0.1.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7fdc5829500425618d4f3bd7798a86e1b66ae5394d72ac4cf0cfe951bee56cc3 |
|
MD5 | d5e728f23016d8ba3cea91c59367acc4 |
|
BLAKE2b-256 | bb60fe147e106983267646bd00cbbe47e6e279f276150845ae44a9f4fb8aa9f5 |