s478839/ium_478839
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Zaktualizuj 'ml_pytorch_mlflow.py'

Browse Source
This commit is contained in:
Sebastian Wałęsa 2022-05-15 23:14:10 +02:00
parent ccff4dcfad
commit a083d9ee7a
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ml_pytorch_mlflow.py
View File

@ -142,7 +142,8 @@ def my_main(epochs):
# inputss, targetss = val_ds # inputss, targetss = val_ds
inputss = pd.DataFrame(inputss, dtype=np.float64) inputss = pd.DataFrame(inputss, dtype=np.float64)
inputss = inputss.to_numpy() # inputss = inputss.to_numpy()
inputsss = inputss.values
MSE = mean_squared_error(expected, predicted) MSE = mean_squared_error(expected, predicted)
MAE = mean_absolute_error(expected, predicted) MAE = mean_absolute_error(expected, predicted)
@ -156,10 +157,10 @@ def my_main(epochs):
input_, target = val_ds[i] input_, target = val_ds[i]
file.write(str(predict_single(input_, target, model))) file.write(str(predict_single(input_, target, model)))
print(inputss) print(inputsss)
input_example = inputss[0] input_example = inputsss[0]
signature = infer_signature(inputss, expected) signature = infer_signature(inputsss, expected)
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
if tracking_url_type_store != "file": if tracking_url_type_store != "file":