diff options
Diffstat (limited to 'become_yukarin/config.py')
| -rw-r--r-- | become_yukarin/config.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/become_yukarin/config.py b/become_yukarin/config.py index 50694b7..f74c83e 100644 --- a/become_yukarin/config.py +++ b/become_yukarin/config.py @@ -1,5 +1,6 @@ import json from pathlib import Path +from typing import List from typing import NamedTuple from typing import Union @@ -14,6 +15,7 @@ class DatasetConfig(NamedTuple): input_var_path: Path target_mean_path: Path target_var_path: Path + features: List[str] seed: int num_test: int @@ -80,6 +82,7 @@ def create_from_json(s: Union[str, Path]): input_var_path=Path(d['dataset']['input_var_path']).expanduser(), target_mean_path=Path(d['dataset']['target_mean_path']).expanduser(), target_var_path=Path(d['dataset']['target_var_path']).expanduser(), + features=d['dataset']['features'], seed=d['dataset']['seed'], num_test=d['dataset']['num_test'], ), |