s464913/ium_464913
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

IUM_05

Browse Source
This commit is contained in:
Mateusz 2024-04-14 14:49:30 +02:00
parent ea02f9dc7b
commit ffdbe0a365
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
predict.py
View File

@ -5,6 +5,7 @@ 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 from sklearn.metrics import confusion_matrix
import numpy as np
def main(): def main():
@ -13,7 +14,8 @@ def main():
y_test = pd.read_csv("data/y_test.csv") y_test = pd.read_csv("data/y_test.csv")
y_pred = model.predict(X_test) y_pred = model.predict(X_test)
y_pred = y_pred > 0.5 y_pred = y_pred >= 0.5
np.savetxt("data/y_pred.csv", y_pred, delimiter=",")
cm = confusion_matrix(y_test, y_pred) cm = confusion_matrix(y_test, y_pred)
print( print(