def discretization_test(raw, filename): with open(filename, "w") as f: f.write("n;precision;recall;accuracy;f1\n") dataset = DataSet(raw) for i in [5, 7, 10, 15, 20, 40]: dataset.discretize_values(i) res = cross_validation(dataset) res = [str(i)] + [str(a) for a in res] f.write(";".join(res) + "\n")
def rules_generated_test(raw, filename): with open(filename, "w") as f: f.write("n;rules\n") dataset = DataSet(raw) for i in [3, 4, 5, 6, 7, 8, 9, 10]: dataset.discretize_values(i) results = [] for j in range(3): # liczba prób values = dataset.cross_validation(5) for v in values: bayes = ILA(v["train"]) results.append(len(bayes.rules)) f.write(str(i) + ";" + str(mean(results)) + "\n")