|
24 | 24 | from google_cloud_pipeline_components.preview.automl.tabular.tabnet_trainer import tabnet_trainer as TabNetTrainerOp |
25 | 25 | from google_cloud_pipeline_components.preview.automl.tabular.utils import get_tabnet_hyperparameter_tuning_job_pipeline_and_parameters |
26 | 26 | from google_cloud_pipeline_components.preview.automl.tabular.utils import get_tabnet_trainer_pipeline_and_parameters |
27 | | -from google_cloud_pipeline_components.preview.automl.tabular.utils import get_wide_and_deep_hyperparameter_tuning_job_pipeline_and_parameters |
28 | | -from google_cloud_pipeline_components.preview.automl.tabular.utils import get_wide_and_deep_trainer_pipeline_and_parameters |
29 | | -from google_cloud_pipeline_components.preview.automl.tabular.wide_and_deep_hyperparameter_tuning_job import wide_and_deep_hyperparameter_tuning_job as WideAndDeepHyperparameterTuningJobOp |
30 | | -from google_cloud_pipeline_components.preview.automl.tabular.wide_and_deep_trainer import wide_and_deep_trainer as WideAndDeepTrainerOp |
31 | 27 | from kfp import components |
32 | 28 |
|
33 | 29 | __all__ = [ |
|
37 | 33 | 'FeatureTransformEngineOp', |
38 | 34 | 'TabNetHyperparameterTuningJobOp', |
39 | 35 | 'TabNetTrainerOp', |
40 | | - 'WideAndDeepHyperparameterTuningJobOp', |
41 | | - 'WideAndDeepTrainerOp', |
42 | 36 | 'get_tabnet_hyperparameter_tuning_job_pipeline_and_parameters', |
43 | 37 | 'get_tabnet_trainer_pipeline_and_parameters', |
44 | | - 'get_wide_and_deep_hyperparameter_tuning_job_pipeline_and_parameters', |
45 | | - 'get_wide_and_deep_trainer_pipeline_and_parameters', |
46 | 38 | ] |
47 | 39 |
|
48 | 40 | tabnet_trainer_pipeline = components.load_component_from_file( |
49 | 41 | # Note, please don't name it as `component.yaml` which will conflict with |
50 | 42 | # the generated file. |
51 | 43 | os.path.join(os.path.dirname(__file__), 'tabnet_trainer_pipeline.yaml') |
52 | 44 | ) |
53 | | - |
54 | | -wide_and_deep_trainer_pipeline = components.load_component_from_file( |
55 | | - # Note, please don't name it as `component.yaml` which will conflict with |
56 | | - # the generated file. |
57 | | - os.path.join( |
58 | | - os.path.dirname(__file__), 'wide_and_deep_trainer_pipeline.yaml' |
59 | | - ) |
60 | | -) |
0 commit comments