added code to preditct data

This commit is contained in:
Mateusz Kuc 2023-06-30 19:13:26 +02:00
parent d4857c3e4b
commit 4a4c809aa9
1 changed files with 18 additions and 0 deletions

18
predict.py Normal file
View File

@ -0,0 +1,18 @@
import tensorflow as tf
import pandas as pd
# Wczytanie danych
data = pd.read_csv('wines.csv')
# Przygotowanie danych wejściowych
input_data = data[['price', 'points']] # Wejście modelu (kolumny 'price' i 'points')
# Wczytanie wytrenowanego modelu
model = tf.keras.models.load_model('trained_model_tf.h5')
# Dokonanie predykcji na danych
predictions = model.predict(input_data)
# Zapisanie wyników predykcji do pliku CSV
output_df = pd.DataFrame(predictions, columns=['predicted_price'])
output_df.to_csv('predictions.csv', index=False)