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
« prev ^ index » next coverage.py v7.4.4, created at 2024-04-21 13:49 -0400
1"""
2Contextualized Regression models.
3"""
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
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}