This commit is contained in:
AWieczarek 2024-06-11 22:50:11 +02:00
parent 42f90164aa
commit 490b8cf773
2 changed files with 1 additions and 1 deletions

Binary file not shown.

View File

@ -43,7 +43,7 @@ def run_experiment(epochs, batch_size, _run):
model.fit(X_train_pad, y_train, epochs=epochs, batch_size=batch_size, validation_split=0.1)
model.save('beer_review_sentiment_model.keras')
_run.add_artifact('beer_review_model.h5')
_run.add_artifact('beer_review_sentiment_model.keras')
test_data = pd.read_csv('beer_reviews_test.csv')
X_test = test_data[['review_aroma', 'review_appearance', 'review_palate', 'review_taste']]