Coverage for contextualized/regression/__init__.py: 100%

10 statements  

« prev     ^ index     » next       coverage.py v7.4.4, created at 2024-04-21 13:49 -0400

1""" 

2Contextualized Regression models. 

3""" 

4 

5from contextualized.regression.datasets import ( 

6 MultivariateDataset, 

7 UnivariateDataset, 

8 MultitaskMultivariateDataset, 

9 MultitaskUnivariateDataset, 

10) 

11from contextualized.regression.losses import MSE, BCELoss 

12from contextualized.regression.regularizers import REGULARIZERS 

13from contextualized.regression.lightning_modules import ( 

14 NaiveContextualizedRegression, 

15 ContextualizedRegression, 

16 MultitaskContextualizedRegression, 

17 TasksplitContextualizedRegression, 

18 ContextualizedUnivariateRegression, 

19 TasksplitContextualizedUnivariateRegression, 

20) 

21from contextualized.regression.trainers import RegressionTrainer 

22 

23DATASETS = { 

24 "multivariate": MultivariateDataset, 

25 "univariate": UnivariateDataset, 

26 "multitask_multivariate": MultitaskMultivariateDataset, 

27 "multitask_univariate": MultitaskUnivariateDataset, 

28} 

29LOSSES = {"mse": MSE, "bceloss": BCELoss} 

30MODELS = ["multivariate", "univariate"] 

31METAMODELS = ["simple", "subtype", "multitask", "tasksplit"] 

32TRAINERS = {"regression_trainer": RegressionTrainer}