-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfigs.py
43 lines (41 loc) · 1.39 KB
/
configs.py
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
# flake8: noqa
import os
from pathlib import Path
from constants import Constants
res_path = Path(os.getcwd()) / Constants.stats_dir
init_file_path = Path(os.getcwd()) / Constants.init_file
temp_post_req_data_path = Path(os.getcwd()) / Constants.post_req_data_file
configs = {
"stats_dir": Constants.stats_dir,
"model_file": Constants.model_file,
"results_path": res_path,
"default_model_path": res_path / Constants.model_file,
"description_file": res_path / Constants.description_file,
"evaluation_file": res_path / Constants.evaluation_file,
"prediction_file": res_path / Constants.prediction_file,
"init_file_path": init_file_path,
"dataset_props": {
"type": "csv",
"split": {"test_size": 0.1, "shuffle": True},
"preprocess": {
"missing_values": "mean",
"scale": {"method": "standard", "target": "inputs"},
},
},
"model_props": {"type": "classification", "algorithm": "NeuralNetwork"},
"available_dataset_props": {
"type": "csv",
"separator": ",",
"split": {"test_size": None, "shuffle": False, "stratify": None},
"preprocess": {
"missing_values": "mean",
"encoding": None,
"scale": None,
},
},
"available_model_props": {
"type": "regression",
"algorithm": "linear regression",
"arguments": "default",
},
}