This commit is contained in:
parent
1060ec9976
commit
09fd3c44ab
@ -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}'"
|
||||
|
@ -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,
|
||||
|
Loading…
Reference in New Issue
Block a user