Tpot

Regression

from tpot import TPOTRegressor

tpot = TPOTRegressor(

generations=None,

population_size=10,

offspring_size=None,

mutation_rate=0.9,

crossover_rate=0.1,

scoring="r2",

cv=5,

subsample=1.0,

n_jobs=-1,

max_time_mins=10,

max_eval_time_mins=2,

random_state=None,

config_dict=None,

template=None,

warm_start=False,

memory=None,

use_dask=True,

periodic_checkpoint_folder=None,

early_stop=5,

verbosity=2,

disable_update_check=False,

)

tpot.fit(X_train, y_train)

yhat = tpot.predict(X_test)

tpot.export('tpot_pipeline.py')

%load tpot_pipeline.py