mirror of
https://github.com/twitter/the-algorithm-ml.git
synced 2024-12-24 07:11:49 +01:00
15 lines
502 B
Python
15 lines
502 B
Python
|
from tml.core.config import base_config
|
||
|
|
from tml.projects.twhin.data.config import TwhinDataConfig
|
||
|
|
from tml.projects.twhin.models.config import TwhinModelConfig
|
||
|
|
from tml.core.config.training import RuntimeConfig, TrainingConfig
|
||
|
|
|
||
|
|
import pydantic
|
||
|
|
|
||
|
|
|
||
|
|
class TwhinConfig(base_config.BaseConfig):
|
||
|
|
runtime: RuntimeConfig = pydantic.Field(RuntimeConfig())
|
||
|
|
training: TrainingConfig = pydantic.Field(TrainingConfig())
|
||
|
|
model: TwhinModelConfig
|
||
|
|
train_data: TwhinDataConfig
|
||
|
|
validation_data: TwhinDataConfig
|