import pandas as pd
from tensorflow import keras
import csv

model = keras.models.load_model('model.h5')
X_test = pd.read_csv('X_test.csv').values
X_train = pd.read_csv('X_train.csv').values
X_val = pd.read_csv('X_val.csv').values

with open('test_predictions.csv', "a") as file:
  writer = csv.writer(file)
  predictions = model.predict(X_test)
  writer.writerow(['PredictedPriceAboveMedian'])
  for pred in predictions:
      result = [0]
      if pred[0] > 0.5:
          result = [1]
      writer.writerow(result)

with open('train_predictions.csv', "a") as file:
  writer = csv.writer(file)
  predictions = model.predict(X_train)
  writer.writerow(['PredictedPriceAboveMedian'])
  for pred in predictions:
      result = [0]
      if pred[0] > 0.5:
          result = [1]
      writer.writerow(result)

with open('dev_predictions.csv', "a") as file:
  writer = csv.writer(file)
  predictions = model.predict(X_val)
  writer.writerow(['PredictedPriceAboveMedian'])
  for pred in predictions:
      result = [0]
      if pred[0] > 0.5:
          result = [1]
      writer.writerow(result)