s464979/ium_464979
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

IUM_06

Browse Source
This commit is contained in:
AWieczarek 2024-05-05 19:01:35 +02:00
parent 4a468eac34
commit 3f4ada3b12
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
IUM_05-model.py
View File

@ -1,5 +1,6 @@
import pandas as pd
import tensorflow as tf
import sys
train_data = pd.read_csv('./beer_reviews_train.csv')
X_train = train_data[['review_aroma', 'review_appearance', 'review_palate', 'review_taste']]
@ -22,6 +23,6 @@ model.compile(optimizer='adam',
loss='binary_crossentropy',
metrics=['accuracy'])
model.fit(X_train_pad, y_train, epochs=40, batch_size=32, validation_split=0.1)
model.fit(X_train_pad, y_train, epochs=sys.argv[1], batch_size=sys.argv[2], validation_split=0.1)
model.save('beer_review_sentiment_model.h5')

11
Jenkinsfile vendored
View File

@ -1,10 +1,13 @@
pipeline {
agent any
triggers {
upstream(upstreamProjects: 'z-s464979-create-datase', threshold: hudson.model.Result.SUCCESS)
}
parameters {
string(name: 'CUTOFF', defaultValue: '10000', description: 'Liczba wierszy do obcięcia ze zbioru danych')
string(name: 'KAGGLE_USERNAME', defaultValue: '', description: 'Kaggle username')
password(name: 'KAGGLE_KEY', defaultValue: '', description: 'Kaggle API key')
string(name: 'EPOCHS', defaultValue: '40', description: 'Number of epochs')
string(name: 'BATCH_SIZE', defaultValue: '32', description: 'Batch size')
buildSelector(defaultSelector: lastSuccessful(), description: 'Which build to use for copying artifacts', name: 'BUILD_SELECTOR')
}
@ -28,7 +31,7 @@ pipeline {
}
steps {
sh "chmod +x ./IUM_05-model.py"
sh "python3 ./IUM_05-model.py"
sh "python3 ./IUM_05-model.py ${params.EPOCHS} ${params.BATCH_SIZE}"
archiveArtifacts artifacts: 'beer_review_sentiment_model.h5', onlyIfSuccessful: true
}
}