tempor.models.clairvoyance2.datasets.dummy module¶

class tempor.models.clairvoyance2.datasets.dummy.DummyDatasetGenerator(n_samples: int, temporal_covariates_n_features: int, temporal_covariates_max_len: int, temporal_covariates_missing_prob: float, static_covariates_n_features: int, static_covariates_missing_prob: float, temporal_targets_n_features: int, temporal_targets_n_categories: int | None, temporal_treatments_n_features: int, temporal_treatments_n_categories: int | None)[source]¶

Bases: object

generate(random_seed: int) → Dataset[source]¶

tempor.models.clairvoyance2.datasets.dummy.dummy_dataset(n_samples: int = 100, temporal_covariates_n_features: int = 5, temporal_covariates_max_len: int = 20, temporal_covariates_missing_prob: float = 0.1, static_covariates_n_features: int = 0, static_covariates_missing_prob: float = 0.0, temporal_targets_n_features: int = 0, temporal_targets_n_categories: int | None = None, temporal_treatments_n_features: int = 0, temporal_treatments_n_categories: int | None = None, random_seed: int = 12345) → Dataset[source]¶