6b8c43b7c9
Some checks failed
s434704-training/pipeline/head There was a failure building this commit
26 lines
703 B
Python
26 lines
703 B
Python
import sys
|
|
import pandas as pd
|
|
import numpy as np
|
|
import tensorflow as tf
|
|
import os.path
|
|
|
|
from tensorflow import keras
|
|
from tensorflow.keras import layers
|
|
from tensorflow.keras.layers.experimental import preprocessing
|
|
|
|
# Wczytanie danych
|
|
test_data = pd.read_csv("./MoviesOnStreamingPlatforms_updated.test")
|
|
|
|
columns_to_use = ['Year', 'Runtime', 'Netflix']
|
|
test_X = tf.convert_to_tensor(test_data[columns_to_use])
|
|
test_Y = tf.convert_to_tensor(test_data[["IMDb"]])
|
|
|
|
model = tf.keras.models.load_model('linear_regression.h5')
|
|
|
|
scores = model.evaluate(x=test_X,
|
|
y=test_Y)
|
|
|
|
with open('rmse.txt', 'w') as file:
|
|
for idx, score in enumerate(scores):
|
|
file.write(str(score))
|