mirror of
https://github.com/open-metadata/OpenMetadata.git
synced 2025-10-31 10:39:30 +00:00
a7c8056bf5
* Prepare infra * Store experiment information in MySQL & MinIO * Use CreateMlModelEntityRequest instead of MlModel for PUT operations * Update MlFlow infra * Prepare MlFlow source * Prepare Mlflow workflow * Simplify test and prepare README * Revert compose * Fix compose * Prepare warnings and fix features * Use non-default port for integration test * Use mlflow-skinny for the client application
23 lines
504 B
Python
23 lines
504 B
Python
"""
|
|
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()
|