INNER CODE UNIT · Python
main
ankonzoid/LearningX · advanced_ML/model_tree/run_model_tree.py:12
def main():
# ====================
# Settings
# ====================
mode = "regr" # "clf" / "regr"
save_model_tree = True # save model tree?
save_model_tree_predictions = True # save model tree predictions/explanations?
cross_validation = True # cross-validate model tree?
# ====================
# Load data
# ====================
data_csv_data_filename = os.path.join("data", "data_clf.csv")
X, y, header = load_csv_data(data_csv_data_filename, mode=mode, verbose=True)
# *********************************************
#
# Insert your models here!