INNER CODE UNIT · Python

save_model_tree_predictions

ankonzoid/LearningX · advanced_ML/model_tree/run_model_tree.py:18

    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!
    #
    # All models must have the following class instantiations:
    #
    #   fit(X, y)
    #   predict(X)
    #   loss(X, y, y_pred)

View source record →