s434749
/
ium_434749
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix
s434749-training/pipeline/head This commit looks good Details

This commit is contained in:
Alagris 2021-05-23 19:46:14 +02:00
parent 1060ec9976
commit 09fd3c44ab
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train-model.Jenkinsfile
View File

@ -10,7 +10,7 @@ pipeline {
stages {
stage('Build') {
steps {
cleanWs()
// cleanWs()
git 'https://git.wmi.amu.edu.pl/s434749/ium_434749.git'
copyArtifacts fingerprintArtifacts: true, projectName: 's434749-create-dataset', selector: lastSuccessful()
sh "python3 train_model.py with 'batch_size=${params.batch_size}' 'learning_rate=${params.learning_rate}' 'epochs=${params.epochs}'"

4
train_model.py
View File

@ -193,10 +193,10 @@ def cfg():
def signature(model,in_alphabet,max_len):
mock_x = [('abc', 'xyz'), ('hey', 'man')]
mock_x = [('abc', 'xyz')]
mock_text, _ = encode_str(mock_x, in_alphabet, max_len)
mock_y = model(mock_text)
return mlflow.models.signature.infer_signature(mock_x, mock_y)
return mlflow.models.signature.infer_signature(mock_text, mock_y)
@ex.automain
def run(kernel_size, hidden_layers, data_file, epochs, teacher_forcing_probability, learning_rate, batch_size, max_len,