Zaktualizuj 'train-mlflow.py'

This commit is contained in:
Kornelia Girejko 2022-05-15 15:15:38 +02:00
parent c71170ef92
commit aceb722599

View File

@ -100,7 +100,7 @@ def my_main(epochs):
#result.write(f'{y_predicted}')
input_example = X_testing[0].numpy()
siganture = pd.DataFrame(X_training, y_predicted_train)
siganture = pd.DataFrame([X_training, y_predicted_train])
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
if tracking_url_type_store != "file":