INNER CODE UNIT · Python
generate_csv_data
ankonzoid/LearningX · advanced_ML/model_tree/run_tests.py:80
def generate_csv_data(func, output_csv_filename, x_range=(0, 1), N=500):
x_vec = np.linspace(x_range[0], x_range[1], N)
y_vec = np.vectorize(func)(x_vec)
with open(output_csv_filename, "w") as f:
writer = csv.writer(f)
field_names = ["x1", "y"]
writer.writerow(field_names)
for (x, y) in zip(x_vec, y_vec):
field_values = [x, y]
writer.writerow(field_values)
def run_tests(ModelTree, data_csv_filename):
print("Running model tree tests...")
eps = 1E-6 # tolerance for test acceptance
X, y, header = load_csv_data(data_csv_filename, mode="regr")
# Test 1