This commit is contained in:
Mateusz 2024-05-04 16:39:48 +02:00
parent 3f95fa102c
commit df42bfcee0
3 changed files with 1 additions and 6 deletions

View File

@ -9,10 +9,6 @@ def main():
build_number = sys.argv[1] build_number = sys.argv[1]
cm = confusion_matrix(y_test, y_pred) cm = confusion_matrix(y_test, y_pred)
print(
"Recall metric in the testing dataset: ",
cm[1, 1] / (cm[1, 0] + cm[1, 1]),
)
accuracy = cm[1, 1] / (cm[1, 0] + cm[1, 1]) accuracy = cm[1, 1] / (cm[1, 0] + cm[1, 1])
with open(r"evaluation/metrics.txt", "a") as f: with open(r"evaluation/metrics.txt", "a") as f:

View File

@ -1,4 +1,3 @@
import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
@ -15,6 +14,7 @@ def main():
plt.xlabel("Build Number") plt.xlabel("Build Number")
plt.ylabel("Accuracy") plt.ylabel("Accuracy")
plt.title("Accuracy of the model over time") plt.title("Accuracy of the model over time")
plt.xticks(range(min(build_numbers), max(build_numbers) + 1))
plt.show() plt.show()
plt.savefig("evaluation/accuracy.png") plt.savefig("evaluation/accuracy.png")

View File

@ -4,7 +4,6 @@ os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
from keras.models import load_model from keras.models import load_model
import pandas as pd import pandas as pd
from sklearn.metrics import confusion_matrix
import numpy as np import numpy as np