INNER CODE UNIT · Python
main
ankonzoid/LearningX · advanced_ML/model_tree/run_tests.py:14
def main():
# ====================
# Run sanity checks on model tree before training (using our own data)
# 1) Reproduce model result on depth-0 model tree
# 2) Reproduce sklearn DecisionTreeRegressor result using mean regression + mse
# 3) Reproduce sklearn DecisionTreeClassifier result using modal class + gini loss
# ====================
run_tests(ModelTree, os.path.join("data", "data_clf.csv"))
# ====================
# For 1D polynomial data using a model tree with linear regression model
# ====================
# Generate 1D polynomial data and save as a csv
func = lambda x: (x-1)*(x-4)*(x-8)*(x-8)
data_csv_data_filename = os.path.join("data", "data_poly4_regr.csv")
generate_csv_data(func, data_csv_data_filename, x_range=(0, 10), N=500)