2024-04-14 13:36:50 +02:00
|
|
|
import os
|
|
|
|
|
|
|
|
os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
|
|
|
|
|
|
|
|
from keras.models import load_model
|
|
|
|
import pandas as pd
|
2024-04-14 14:49:30 +02:00
|
|
|
import numpy as np
|
2024-04-14 13:36:50 +02:00
|
|
|
|
|
|
|
|
|
|
|
def main():
|
|
|
|
model = load_model("model/model.keras")
|
|
|
|
X_test = pd.read_csv("data/X_test.csv")
|
|
|
|
|
|
|
|
y_pred = model.predict(X_test)
|
2024-04-14 14:49:30 +02:00
|
|
|
y_pred = y_pred >= 0.5
|
2024-04-14 13:36:50 +02:00
|
|
|
|
2024-05-04 15:25:54 +02:00
|
|
|
os.makedirs("evaluation", exist_ok=True)
|
|
|
|
np.savetxt("evaluation/y_pred.csv", y_pred, delimiter=",")
|
2024-04-14 13:36:50 +02:00
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
main()
|