added code to preditct data
This commit is contained in:
parent
d4857c3e4b
commit
4a4c809aa9
18
predict.py
Normal file
18
predict.py
Normal 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)
|
Loading…
Reference in New Issue
Block a user