|
194 | 194 | " parser.add_argument(\"--n_estimators\", required=False, default=100, type=int)\n", |
195 | 195 | " parser.add_argument(\"--learning_rate\", required=False, default=0.1, type=float)\n", |
196 | 196 | " parser.add_argument(\"--registered_model_name\", type=str, help=\"model name\")\n", |
197 | | - " parser.add_argument(\"--model\", type=str, default="./outputs", help=\"output model directory\")\n", |
| 197 | + " parser.add_argument(\"--model\", type=str, default=\"./outputs\", help=\"output model directory\")\n", |
198 | 198 | " args = parser.parse_args()\n", |
199 | 199 | " \n", |
200 | 200 | " # Start Logging\n", |
|
368 | 368 | " learning_rate=0.25,\n", |
369 | 369 | " registered_model_name=registered_model_name,\n", |
370 | 370 | " ),\n", |
371 | | - " Outputs=dict(\n", |
372 | | - " model=Output(type="uri_folder")\n", |
373 | | - " ),\n", |
| 371 | + " Outputs=dict(\n", |
| 372 | +" model=Output(type=\"uri_folder\")\n", |
| 373 | +" ),\n", |
374 | 374 | " code=\"./src/\", # location of source code\n", |
375 | 375 | " command=\"python main.py --data ${{inputs.data}} --test_train_ratio ${{inputs.test_train_ratio}} --learning_rate ${{inputs.learning_rate}} --registered_model_name ${{inputs.registered_model_name}}\",\n", |
376 | 376 | " environment=\"azureml://registries/azureml/environments/sklearn-1.5/labels/latest\",\n", |
|
0 commit comments