-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.yaml
29 lines (29 loc) · 809 Bytes
/
config.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
main:
tracking_uri: "http://127.0.0.1:5000"
experiment_name: "development"
steps: "all"
download:
uri: "https://archive.ics.uci.edu/ml/machine-learning-databases/iris/iris.data"
file_name: "iris.data"
s3_path: ""
clean:
raw_data: "../get_data/data/iris.data"
file_name: "iris.csv"
col_names: "sepal_length,sepal_width,petal_length,petal_width,species"
s3_path: ""
split:
clean_data: "../clean_data/data/iris.csv"
test_size: 0.3
random_seed: 42
file_names: "iris_train.csv,iris_test.csv"
s3_path: ""
train:
train_data: "../split_data/data/iris_train.csv"
target: "species"
model_config: "model_config.json"
model_name: "model.joblib"
s3_path: ""
test:
test_data: "../split_data/data/iris_test.csv"
target: "species"
model_path: "../train_model/model/model.joblib"