MLJAR AutoML modes

mljar_automl_mode.py

'''
MLJAR AutoML can work in 4 modes:
- Explain
- Perform
- Compete
- Optuna
'''

# training in Explain mode, perfect for data exploration
automl = AutoML(mode="Explain")
automl.fit(X, y)

# training in Perform mode, perfect for production-ready systems
# AutoML trained for 4 hours
automl = AutoML(mode="Perform", total_time_limit=4*3600)
automl.fit(X, y)

# training in Compete mode, perfect for high performing ML pipelines
automl = AutoML(mode="Compete", total_time_limit=4*3600)
automl.fit(X, y)


# training in Optuna mode, highly tune selected algorithm
automl = AutoML(
    mode="Optuna", 
    algorithms=["CatBoost", "LightGBM", "Xgboost"],
    optuna_time_budget=3600,
    eval_metric="f1"
)
automl.fit(X, y)