Update 'evaluate.py'

2022-04-25 23:27:04 +02:00 · 2022-04-25 23:27:04 +02:00 · 3c26863f73
commit 3c26863f73
parent 0482fe10bc
1 changed files with 2 additions and 4 deletions
--- a/evaluate.py
+++ b/evaluate.py
@ -20,13 +20,11 @@ test_results = model.evaluate(

 # Zapis wartości liczbowej metryki do pliku
 with open('eval_results.txt', 'a+') as f:
-    f.write(str(test_results))
+    f.write(str(test_results) + '\n')

 # Wygenerowanie i zapisanie do pliku wykresu
 with open('eval_results.txt') as f:
-    scores = []
-    for line in f:
-        scores.append(float(line))
+    scores = [float(line) for line in f if line]
    builds = list(range(1, len(scores) + 1))

    plot = plt.plot(builds, scores)