ium_z487175/DL-prediction.py

21 lines
453 B
Python

import pickle
import numpy as np
with open('model_with_data.pickle', 'rb') as file:
loaded_data = pickle.load(file)
# Wczytanie modelu
model = loaded_data[0]
#Wczytanie danych testowych
X_test_scaled = loaded_data[3]
y_test_encoded = loaded_data[4]
# Predykcja
y_pred = model.predict(X_test_scaled)
y_pred_classess = np.argmax(y_pred, axis=1)
# Zapisanie wyników predykcji do pliku
np.savetxt('results_prediction.csv', y_pred, delimiter=',')