27 lines
555 B
Python
27 lines
555 B
Python
|
import os
|
||
|
|
|
||
|
|
os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
|
||
|
|
|
||
|
|
from keras.models import load_model
|
||
|
|
import pandas as pd
|
||
|
|
from sklearn.metrics import confusion_matrix
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
model = load_model("model/model.keras")
|
||
|
|
X_test = pd.read_csv("data/X_test.csv")
|
||
|
|
y_test = pd.read_csv("data/y_test.csv")
|
||
|
|
|
||
|
|
y_pred = model.predict(X_test)
|
||
|
|
y_pred = y_pred > 0.5
|
||
|
|
|
||
|
|
cm = confusion_matrix(y_test, y_pred)
|
||
|
|
print(
|
||
|
|
"Recall metric in the testing dataset: ",
|
||
|
|
cm[1, 1] / (cm[1, 0] + cm[1, 1]),
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|