23 lines
504 B
Python
Raw Normal View History

"""
This is a simple script for testing the environment.
You can run it as `python experiment.py`, and a new
experiment should appear in the remote server.
"""
import mlflow
# Create experiment to the remote server
mlflow_uri = "http://localhost:5000"
mlflow.set_tracking_uri(mlflow_uri)
mlflow.set_experiment("mlflow_test1")
mlflow.log_param("param1", "value1")
mlflow.log_param("param2", "value2")
for i in range(10):
mlflow.log_metric("metric1", value=1 / (i + 1), step=i)
mlflow.end_run()