add new model

This commit is contained in:
Michael Duong 2023-10-10 12:32:25 +00:00
parent ad2c266727
commit 9e7d0fa538
2 changed files with 22 additions and 22 deletions

View file

@ -13,6 +13,6 @@ default:
output_filepath: ./data/model/allmodels/
problem_type: regression
eval_metric: mean_squared_error #mean_absolute_error
time_limit: 1000
time_limit: 4000
presets: medium_quality
excluded_model_types: ['KNN', 'RF']

View file

@ -29,8 +29,8 @@ stages:
outs:
- path: data/prepared_data/
hash: md5
md5: 71e63a792f7723e2aea0709efde1a92b.dir
size: 31751660
md5: e0be70d5025e40dd0d655d9949f72130.dir
size: 31800776
nfiles: 2
build_model:
cmd: python 2_build_model.py
@ -41,8 +41,8 @@ stages:
size: 5359
- path: data/prepared_data
hash: md5
md5: 71e63a792f7723e2aea0709efde1a92b.dir
size: 31751660
md5: e0be70d5025e40dd0d655d9949f72130.dir
size: 31800776
nfiles: 2
params:
configs/build_model.yaml:
@ -58,7 +58,7 @@ stages:
output_filepath: ./data/model/allmodels/
problem_type: regression
eval_metric: mean_squared_error
time_limit: 1000
time_limit: 4000
presets: medium_quality
excluded_model_types:
- KNN
@ -66,13 +66,13 @@ stages:
outs:
- path: data/model/
hash: md5
md5: 0ffc51be7c8381c9e4106309e3e05ca3.dir
size: 345904743
md5: 14ca33cde5e86770135f768abaf84978.dir
size: 422447808
nfiles: 27
- path: metrics/fit_metrics.json
hash: md5
md5: 3d4ff3a3ca3c327e2c1e9aa1338c18ce
size: 220
md5: 41bfb8d2da8f06d1864d73ce125cc6aa
size: 221
generate_predictions:
cmd: python 3_generate_predictions.py
deps:
@ -82,13 +82,13 @@ stages:
size: 3028
- path: data/model
hash: md5
md5: 0ffc51be7c8381c9e4106309e3e05ca3.dir
size: 345904743
md5: 14ca33cde5e86770135f768abaf84978.dir
size: 422447808
nfiles: 27
- path: data/prepared_data
hash: md5
md5: 71e63a792f7723e2aea0709efde1a92b.dir
size: 31751660
md5: e0be70d5025e40dd0d655d9949f72130.dir
size: 31800776
nfiles: 2
params:
configs/settings.yaml:
@ -100,8 +100,8 @@ stages:
outs:
- path: data/predictions/
hash: md5
md5: 00ff804016290d56e1490e59c098b060.dir
size: 351811
md5: 40d0c7a7fd4a15add0615e322cf341a0.dir
size: 352151
nfiles: 1
generate_metrics:
cmd: python 4_generate_metrics.py
@ -112,13 +112,13 @@ stages:
size: 4487
- path: data/predictions
hash: md5
md5: 00ff804016290d56e1490e59c098b060.dir
size: 351811
md5: 40d0c7a7fd4a15add0615e322cf341a0.dir
size: 352151
nfiles: 1
- path: data/prepared_data
hash: md5
md5: 71e63a792f7723e2aea0709efde1a92b.dir
size: 31751660
md5: e0be70d5025e40dd0d655d9949f72130.dir
size: 31800776
nfiles: 2
params:
configs/settings.yaml:
@ -128,8 +128,8 @@ stages:
outs:
- path: metrics/metrics.json
hash: md5
md5: 63ef63e4fabe929b914a0059ceeddabc
size: 221
md5: 4e023650240e78d6ad761f1db7aac922
size: 220
startup_cleanup:
cmd: python 0_startup_cleanup.py
deps: