Zaktualizuj 'train-mlflow.py'
This commit is contained in:
parent
b3584a5545
commit
a21d7d094e
@ -106,7 +106,7 @@ def my_main(epochs):
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
input_example = X_testing[0]
|
input_example = np.array(X_testing[0])
|
||||||
siganture = mlflow.models.signature.infer_signature(np.array(X_training), np.array(y_predicted_train))
|
siganture = mlflow.models.signature.infer_signature(np.array(X_training), np.array(y_predicted_train))
|
||||||
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
|
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user