add dataset classes

1 files changed, 20 insertions, 0 deletions

diff --git a/become_yukarin/config.py b/become_yukarin/config.py
new file mode 100644
index 0000000..b15dc6f
--- /dev/null
+++ b/become_yukarin/config.py
@@ -0,0 +1,20 @@
+from typing import NamedTuple
+
+from .data_struct import AcousticFeature
+from .param import Param
+
+
+class DatasetConfig(NamedTuple):
+    param: Param
+    input_glob: str
+    target_glob: str
+    input_mean: AcousticFeature
+    input_var: AcousticFeature
+    target_mean: AcousticFeature
+    target_var: AcousticFeature
+    seed: int
+    num_test: int
+
+
+class Config(NamedTuple):
+    dataset_config: DatasetConfig