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

fix
Some checks failed
s434749-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Alagris 2021-05-23 20:19:22 +02:00
parent b8340d0604
commit 1dda33aec7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train_model.py
View File

@ -197,7 +197,7 @@ def signature(model, in_alphabet, max_len):
mock_x = [('abc', 'xyz')] mock_x = [('abc', 'xyz')]
mock_text, _ = encode_str(mock_x, in_alphabet, max_len) mock_text, _ = encode_str(mock_x, in_alphabet, max_len)
mock_y = model(mock_text) mock_y = model(mock_text)
return mlflow.models.signature.infer_signature(mock_text, mock_y) return mlflow.models.signature.infer_signature(mock_text.numpy(), mock_y.numpy())
@ex.automain @ex.automain