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")
