ium_464913/predict.py

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
import numpy as np


def main():
    model = load_model("model/model.keras")
    X_test = pd.read_csv("data/X_test.csv")

    y_pred = model.predict(X_test)
    y_pred = y_pred >= 0.5

    os.makedirs("evaluation", exist_ok=True)
    os.system("rm -rf evaluation/*")
    np.savetxt("evaluation/y_pred.csv", y_pred, delimiter=",")


if __name__ == "__main__":
    main()