def create(file_store_path, experiment_name): """ Creates a new experiment in FileStore backend. """ fs = file_store.FileStore(file_store_path) exp_id = fs.create_experiment(experiment_name) print("Created experiment '%s' with id '%d'" % (experiment_name, exp_id))
def create(file_store, artifact_root, experiment_name): """ Creates a new experiment in FileStore backend. """ fs = store.FileStore(file_store, artifact_root) exp_id = fs.create_experiment(experiment_name) print("Created experiment '%s' with id %d" % (experiment_name, exp_id))
def list_experiments(file_store): """ List all experiment in FileStore backend. """ fs = store.FileStore(file_store) experiments = fs.list_experiments() table = [[exp.experiment_id, exp.name, os.path.abspath(exp.artifact_location)] for exp in experiments] print(tabulate(sorted(table), headers=["Experiment Id", "Name", "Artifact Location"]))
def test_docker_project_tracking_uri_propagation( ProfileConfigProvider, tmpdir, tracking_uri, expected_command_segment): # pylint: disable=unused-argument build_docker_example_base_image() mock_provider = mock.MagicMock() mock_provider.get_config.return_value = \ DatabricksConfig("host", "user", "pass", None, insecure=True) ProfileConfigProvider.return_value = mock_provider # Create and mock local tracking directory local_tracking_dir = os.path.join(tmpdir.strpath, "mlruns") if tracking_uri is None: tracking_uri = local_tracking_dir old_uri = mlflow.get_tracking_uri() try: mlflow.set_tracking_uri(tracking_uri) with mock.patch("mlflow.tracking.utils._get_store") as _get_store_mock: _get_store_mock.return_value = file_store.FileStore(local_tracking_dir) mlflow.projects.run(TEST_DOCKER_PROJECT_DIR, experiment_id=0) finally: mlflow.set_tracking_uri(old_uri)