Compare commits

...
Author SHA1 Message Date
Andrew Ferlitsch 604cf5964b fix: scikit-learn version 2023-07-31 20:59:04 +00:00
@@ -568,7 +568,7 @@
"outputs": [],
"source": [
"@component(\n",
" packages_to_install=[\"scikit-learn\"],\n",
" packages_to_install=[\"scikit-learn==1.2.2\"],\n",
" base_image=\"python:3.9\",\n",
" output_component_file=\"wine_classification_component.yaml\",\n",
")\n",
@@ -613,7 +613,7 @@
},
"outputs": [],
"source": [
"@component(packages_to_install=[\"scikit-learn\"], base_image=\"python:3.9\")\n",
"@component(packages_to_install=[\"scikit-learn==1.2.2\"], base_image=\"python:3.9\")\n",
"def iris_sgdclassifier(\n",
" test_samples_fraction: float,\n",
" metricsc: Output[ClassificationMetrics],\n",
@@ -660,7 +660,7 @@
"outputs": [],
"source": [
"@component(\n",
" packages_to_install=[\"scikit-learn\"],\n",
" packages_to_install=[\"scikit-learn==1.2.2\"],\n",
" base_image=\"python:3.9\",\n",
")\n",
"def iris_logregression(\n",
@@ -762,7 +762,7 @@
"\n",
"compiler.Compiler().compile(\n",
" pipeline_func=pipeline,\n",
" package_path=\"tabular classification_pipeline.json\".replace(\" \", \"_\"),\n",
" package_path=\"tabular_classification_pipeline.json\",\n",
")"
]
},
@@ -789,8 +789,8 @@
"\n",
"job = aip.PipelineJob(\n",
" display_name=DISPLAY_NAME,\n",
" template_path=\"tabular classification_pipeline.json\".replace(\" \", \"_\"),\n",
" job_id=f\"tabular classification-v2{UUID}-1\".replace(\" \", \"\"),\n",
" template_path=\"tabular_classification_pipeline.json\",\n",
" job_id=f\"tabular-classification-v2{UUID}-1\",\n",
" pipeline_root=PIPELINE_ROOT,\n",
" parameter_values={\"seed\": 7, \"splits\": 10},\n",
")\n",
@@ -847,8 +847,8 @@
"source": [
"job = aip.PipelineJob(\n",
" display_name=\"iris_\" + UUID,\n",
" template_path=\"tabular classification_pipeline.json\".replace(\" \", \"_\"),\n",
" job_id=f\"tabular classification-pipeline-v2{UUID}-2\".replace(\" \", \"\"),\n",
" template_path=\"tabular_classification_pipeline.json\",\n",
" job_id=f\"tabular-classification-pipeline-v2{UUID}-2\",\n",
" pipeline_root=PIPELINE_ROOT,\n",
" parameter_values={\"seed\": 5, \"splits\": 7},\n",
")\n",
@@ -976,6 +976,8 @@
},
"outputs": [],
"source": [
"import os\n",
"\n",
"delete_bucket = False\n",
"\n",
"job.delete()\n",