Spaces:
Sleeping
Sleeping
File size: 2,321 Bytes
d4f1c9d d4e0e95 d4f1c9d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
import comet_ml
def start_experiment(
comet_api_key,
comet_workspace,
comet_project_name,
comet_experiment_name,
experiment,
):
if comet_api_key is None:
experiment = None
return (
experiment,
"""
Please add your API key in order to log your predictions to a Comet Experiment.
If you don't have a Comet account yet, you can sign up using the link below:
https://www.comet.ml/signup
""",
)
try:
if comet_experiment_name:
# Retrieve the Experiment if it already exists
api_experiment = get_experiment(
{
"api_key": comet_api_key,
"workspace": comet_workspace,
"project_name": comet_project_name,
"experiment": comet_experiment_name,
}
)
else:
# Create a new Experiment
api_experiment = comet_ml.APIExperiment(
api_key=comet_api_key,
workspace=comet_workspace,
project_name=comet_project_name,
)
api_experiment.log_other("Created from", "Spaces")
experiment = {
"api_key": comet_api_key,
"workspace": comet_workspace,
"project_name": comet_project_name,
"experiment": api_experiment.name,
}
return experiment, f"Started {api_experiment.name}. Happy logging!😊"
except Exception as e:
return None, e
def get_experiment(experiment_state):
try:
api_key = experiment_state.get("api_key")
workspace = experiment_state.get("workspace")
project = experiment_state.get("project_name")
experiment_name = experiment_state.get("experiment")
return comet_ml.API(api_key=api_key).get_experiment(
workspace=workspace, project_name=project, experiment=experiment_name
)
except Exception as e:
return None
def get_experiment_status(experiment_state):
experiment = get_experiment(experiment_state)
if experiment is not None:
name = experiment.name
return experiment_state, f"Currently logging to: {name}"
return experiment_state, f"No Experiments found"
|