Slide 43
Slide 43 text
AutoMLの関数化
def pycaret_model(train, target, test, n_select, fold, opt, exclude):
print('Setup Your Data....')
setup(data=train,
target=target,
silent= True)
print('Comparing Models....')
top3 = compare_models(sort=opt, n_select=n_select, fold = fold, exclude = exclude)
print('Blending Models....')
blended = blend_models(estimator_list= top3, fold=fold)
print('Tuning Models....')
tuned_blended = tune_model(blended)
print('Finallizing Models....')
final_model = finalize_model(tuned_blended)
print('Done...!!!')
pred_esb = predict_model(final_model, test)
re = pred_esb['Label'] 43