tempor.models.clairvoyance2.datasets package¶

Subpackages¶

tempor.models.clairvoyance2.datasets.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]¶

tempor.models.clairvoyance2.datasets.simple_pkpd_dataset(n_timesteps: int = 30, time_index_treatment_event: int = 25, n_control_samples: int = 200, n_treated_samples: int = 200, seed: int = 100)[source]¶

tempor.models.clairvoyance2.datasets.uci_diabetes(data_home: str | None = None, refresh_cache: bool = False, redownload: bool = False, make_regular: bool = False, use_int_index: bool = False) → Dataset[source]¶