better model

This commit is contained in:
Michael Duong 2023-09-11 16:57:44 +01:00
parent a2ea03733d
commit 74bce78d2e
3 changed files with 19 additions and 18 deletions

View file

@ -0,0 +1 @@
{}

View file

@ -1,6 +1,6 @@
dataclient_type: minio dataclient_type: minio
data_location: s3://dev_bucket data_location: s3://dev_bucket
train_proportion: 0.4 train_proportion: 0.8
output_train_filepath: ./data/prepared_data/train.parquet output_train_filepath: ./data/prepared_data/train.parquet
output_test_filepath: ./data/prepared_data/test.parquet output_test_filepath: ./data/prepared_data/test.parquet

View file

@ -11,12 +11,12 @@ stages:
configs/prepare_data.yaml: configs/prepare_data.yaml:
output_test_filepath: ./data/prepared_data/test.parquet output_test_filepath: ./data/prepared_data/test.parquet
output_train_filepath: ./data/prepared_data/train.parquet output_train_filepath: ./data/prepared_data/train.parquet
train_proportion: 0.4 train_proportion: 0.8
outs: outs:
- path: data/prepared_data/ - path: data/prepared_data/
hash: md5 hash: md5
md5: 2c91bf9e7db44bddf119efd1e56ade44.dir md5: f0d462fe6b1a856a827409a745539285.dir
size: 37095 size: 36169
nfiles: 2 nfiles: 2
build_model: build_model:
cmd: python build_model.py cmd: python build_model.py
@ -27,8 +27,8 @@ stages:
size: 3576 size: 3576
- path: data/prepared_data - path: data/prepared_data
hash: md5 hash: md5
md5: 2c91bf9e7db44bddf119efd1e56ade44.dir md5: f0d462fe6b1a856a827409a745539285.dir
size: 37095 size: 36169
nfiles: 2 nfiles: 2
params: params:
configs/build_model.yaml: configs/build_model.yaml:
@ -41,7 +41,7 @@ stages:
outs: outs:
- path: data/model/ - path: data/model/
hash: md5 hash: md5
md5: 2d500f958c97a8509fc37e94ab7c01df.dir md5: fb7ae4137b445dc91e840b794d72e940.dir
size: 1096 size: 1096
nfiles: 1 nfiles: 1
generate_predictions: generate_predictions:
@ -49,13 +49,13 @@ stages:
deps: deps:
- path: data/model - path: data/model
hash: md5 hash: md5
md5: 2d500f958c97a8509fc37e94ab7c01df.dir md5: fb7ae4137b445dc91e840b794d72e940.dir
size: 1096 size: 1096
nfiles: 1 nfiles: 1
- path: data/prepared_data - path: data/prepared_data
hash: md5 hash: md5
md5: 2c91bf9e7db44bddf119efd1e56ade44.dir md5: f0d462fe6b1a856a827409a745539285.dir
size: 37095 size: 36169
nfiles: 2 nfiles: 2
- path: generate_predictions.py - path: generate_predictions.py
hash: md5 hash: md5
@ -69,21 +69,21 @@ stages:
outs: outs:
- path: data/predictions/ - path: data/predictions/
hash: md5 hash: md5
md5: deb4baede52e3c3d968b7737390a4f03.dir md5: 4d5854903b25bdae15d99c934ebcfb99.dir
size: 4184 size: 2531
nfiles: 1 nfiles: 1
generate_metrics: generate_metrics:
cmd: python generate_metrics.py cmd: python generate_metrics.py
deps: deps:
- path: data/predictions - path: data/predictions
hash: md5 hash: md5
md5: deb4baede52e3c3d968b7737390a4f03.dir md5: 4d5854903b25bdae15d99c934ebcfb99.dir
size: 4184 size: 2531
nfiles: 1 nfiles: 1
- path: data/prepared_data - path: data/prepared_data
hash: md5 hash: md5
md5: 2c91bf9e7db44bddf119efd1e56ade44.dir md5: f0d462fe6b1a856a827409a745539285.dir
size: 37095 size: 36169
nfiles: 2 nfiles: 2
- path: generate_metrics.py - path: generate_metrics.py
hash: md5 hash: md5
@ -96,5 +96,5 @@ stages:
outs: outs:
- path: metrics/metrics.json - path: metrics/metrics.json
hash: md5 hash: md5
md5: 448cfa9c5f954b50fd944b8e322534af md5: 3c9306e992b07491ff7e642949d6bc47
size: 181 size: 182