feat: add metric and parameter autologging to experiments

PiperOrigin-RevId: 507775772
googleapis · Feb 7, 2023 · 96e9e12 · 96e9e12
1 parent 8613f58
commit 96e9e12
Show file tree

Hide file tree

Showing 13 changed files with 2,024 additions and 23 deletions.
diff --git a/google/cloud/aiplatform/__init__.py b/google/cloud/aiplatform/__init__.py
@@ -94,6 +94,7 @@
 log_model = metadata.metadata._experiment_tracker.log_model
 get_experiment_df = metadata.metadata._experiment_tracker.get_experiment_df
 start_run = metadata.metadata._experiment_tracker.start_run
+autolog = metadata.metadata._experiment_tracker.autolog
 start_execution = metadata.metadata._experiment_tracker.start_execution
 log = metadata.metadata._experiment_tracker.log
 log_time_series_metrics = metadata.metadata._experiment_tracker.log_time_series_metrics

diff --git a/google/cloud/aiplatform/_mlflow_plugin/__init__.py b/google/cloud/aiplatform/_mlflow_plugin/__init__.py
@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2022 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#