kdudzic
/
ium_s449288
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update 'evaluate.py'
s449288-evaluation/pipeline/head There was a failure building this commit Details
s449288-training/pipeline/head There was a failure building this commit Details

This commit is contained in:
Kacper Dudzic 2022-04-25 23:11:03 +02:00
parent 73eb538317
commit f8fa8f2d39
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
evaluate.py
View File

@ -8,6 +8,7 @@ import numpy as np
model = keras.models.load_model('lego_reg_model') model = keras.models.load_model('lego_reg_model')
# Załadowanie zbioru testowego # Załadowanie zbioru testowego
data_test = pd.read_csv('lego_sets_clean_test.csv')
test_piece_counts = np.array(data_test['piece_count']) test_piece_counts = np.array(data_test['piece_count'])
test_prices = np.array(data_test['list_price']) test_prices = np.array(data_test['list_price'])