truefoundry_mlfoundry_quickstart_logging_params_metrics_model.py

1import os
2from mlfoundry import mlfoundry
3
4# Initialize the client
5# Ensure you have set the TFP_API_KEY and TFP_HOST environment variables or provided them here
6client = mlfoundry.get_client()
7
8# Create or get an existing project
9project_name = "my-quickstart-project"
10client.create_project(project_name)
11
12# Start a run to log experiments
13with client.create_run(project_name=project_name, run_name="quickstart-run") as run:
14    # Log hyperparameters
15    run.log_params({"learning_rate": 0.01, "epochs": 10})
16    
17    # Log metrics
18    run.log_metrics({"accuracy": 0.95, "loss": 0.05}, step=1)
19    
20    # Log a dummy model (replace with your actual model object)
21    # run.log_model(model=my_model, name="my-model", framework="sklearn")
22
23print(f"Run logging completed. View your run in the TrueFoundry dashboard.")