Index

A

all_augmented_data (AugmentedBernoulliDatasetConfigs property) all_augmented_data_length (AugmentedBernoulliDatasetConfigs property) all_augmented_test_data (AugmentedBernoulliDatasetConfigs property) all_augmented_test_data_length (AugmentedBernoulliDatasetConfigs property) all_augmented_training_data (AugmentedBernoulliDatasetConfigs property) all_augmented_training_data_length (AugmentedBernoulliDatasetConfigs property) all_augmented_validation_data (AugmentedBernoulliDatasetConfigs property) all_augmented_validation_data_length (AugmentedBernoulliDatasetConfigs property) all_data (AugmentedBernoulliDatasetConfigs property) all_data_length (AugmentedBernoulliDatasetConfigs property) all_test_data (AugmentedBernoulliDatasetConfigs property) all_test_data_length (AugmentedBernoulliDatasetConfigs property) all_training_data (AugmentedBernoulliDatasetConfigs property) all_training_data_length (AugmentedBernoulliDatasetConfigs property) all_validation_data (AugmentedBernoulliDatasetConfigs property) all_validation_data_length (AugmentedBernoulliDatasetConfigs property) assert_monotone() (in module pytorch_utils.utils) assert_monotone_probability() (MonotoneBernoulliProbability method)

augment_data() (DataAugmentationConfig method) augment_transform_to_tensors() (AugmentedBernoulliDataModule method) augmentation_length() (DataAugmentationConfig method) augmented_col (AugmentedBernoulliDataset attribute) (AugmentedBernoulliDatasetConfig property) (AugmentedBernoulliDatasetConfigs property) (DataAugmentationConfig attribute), [1] augmented_data (AugmentedBernoulliDatasetConfig property) augmented_data_length (AugmentedBernoulliDatasetConfig property) augmented_test_data (AugmentedBernoulliDatasetConfig property) augmented_test_data_length (AugmentedBernoulliDatasetConfig property) augmented_training_data (AugmentedBernoulliDatasetConfig property) augmented_training_data_length (AugmentedBernoulliDatasetConfig property) augmented_validation_data (AugmentedBernoulliDatasetConfig property) augmented_validation_data_length (AugmentedBernoulliDatasetConfig property) AugmentedBernoulliDataModule (class in pytorch_utils.data_modules) AugmentedBernoulliDataset (class in pytorch_utils.datasets) AugmentedBernoulliDatasetConfig (class in pytorch_utils.dataset_configurations) AugmentedBernoulliDatasetConfigs (class in pytorch_utils.dataset_configurations)

B

BadCLIParameterException BadConfigException BadConfigFormatException BadConfigLogLevelException BadConfigMissingInputException

BadConfigMissingOutputException BadConfigPathException BadConfigSparkMasterException batch_size (AugmentedBernoulliDataModule property) BatchNorm1dNonNeg (class in pytorch_utils.modules) BiLinearSemiNonNeg (class in pytorch_utils.modules)

C

calibration_curve() (WeightedBinaryCalibrationError method) CatalogError CategoricalFeatureEmbedding (class in pytorch_utils.utils) check_compatibility() (AugmentedBernoulliDatasetConfigs method) class_full_name() (in module pytorch_utils.miscellaneous) clear_data() (AugmentedBernoulliDataModule method) (AugmentedBernoulliDataset method) (AugmentedBernoulliDatasetConfig method) (AugmentedBernoulliDatasetConfigs method) (ProbabilityPredictor method)

columns (AugmentedBernoulliDatasetConfig property) (AugmentedBernoulliDatasetConfigs property) compute() (WeightedBinaryCalibrationError method) configure_optimizers() (MonotoneBernoulliProbability method)

D

data (AugmentedBernoulliDataset attribute) (AugmentedBernoulliDatasetConfig attribute), [1] data_augmentation_config (AugmentedBernoulliDatasetConfig attribute), [1], [2] data_augmentation_scaling_factors (AugmentedBernoulliDataset attribute) (AugmentedBernoulliDatasetConfig property) data_module (ProbabilityPredictor attribute) DataAugmentationConfig (class in pytorch_utils.dataset_configurations)

DataclassType (class in pytorch_utils.miscellaneous) dataframe (AugmentedBernoulliDataset property) DataSetError DataSplitConfig (class in pytorch_utils.dataset_configurations) delta_table (AugmentedBernoulliDatasetConfig attribute) dtypes (AugmentedBernoulliDatasetConfig property) (AugmentedBernoulliDatasetConfigs property)

E

embedding_size (CategoricalFeatureEmbedding attribute)

F

feature_name (CategoricalFeatureEmbedding attribute) fit (MLStage attribute) fit_preprocessing_pipeline() (AugmentedBernoulliDataModule method) fitted_preprocessing_pipeline (AugmentedBernoulliDataset attribute) format_to_tensors() (AugmentedBernoulliDataModule method) forward() (BatchNorm1dNonNeg method) (BiLinearSemiNonNeg method) (LinearNonNeg method) (LinearSemiNonNeg method) (MeanImputationEmbedding method) (MonotoneBernoulliProbability method) (Partitioned method) (ShiftedEmbedding method)

from_config() (AugmentedBernoulliDataset class method) from_meta_dataframe() (AugmentedBernoulliDatasetConfig class method)

G

get_embedding_size() (in module pytorch_utils.utils)

I

InconsistentDatasetConfigurations input_features_dtypes (AugmentedBernoulliDataModule property) InvalidDataFormatError

is_preprocessing_pipeline_fitted (AugmentedBernoulliDataModule property) is_success (AugmentedBernoulliDataset attribute) (AugmentedBernoulliDatasetConfig attribute), [1], [2]

L

label_col (AugmentedBernoulliDataset attribute) labels_dtype (AugmentedBernoulliDataset attribute) learning_rate (MonotoneBernoulliProbability property) LinearNonNeg (class in pytorch_utils.modules) LinearSemiNonNeg (class in pytorch_utils.modules) load_from_checkpoint() (AugmentedBernoulliDataModule class method) (ProbabilityPredictor class method) LocalDirNotWriteableException

log() (AugmentedBernoulliDataModule method) (AugmentedBernoulliDatasetConfig method) (AugmentedBernoulliDatasetConfigs method) (CategoricalFeatureEmbedding method) (DataAugmentationConfig method) logger (AugmentedBernoulliDatasetConfig attribute) (CategoricalFeatureEmbedding attribute) (DataAugmentationConfig attribute) (DataSplitConfig attribute)

M

max_augmented_value (AugmentedBernoulliDataset attribute) max_value (DataAugmentationConfig attribute) MeanImputationEmbedding (class in pytorch_utils.modules) metadata (AugmentedBernoulliDatasetConfig attribute), [1] min_augmented_value (AugmentedBernoulliDataset attribute) min_value (DataAugmentationConfig attribute) MissingConfigException MissingConfigFileException MissingDatasetError MLError MLStage (class in pytorch_utils.datasets) module pytorch_utils pytorch_utils.data_modules pytorch_utils.dataset_configurations pytorch_utils.datasets pytorch_utils.exceptions pytorch_utils.metrics pytorch_utils.miscellaneous pytorch_utils.modules pytorch_utils.utils

module (ProbabilityPredictor attribute) module_scope (MonotoneBernoulliProbability attribute) MonotoneBernoulliProbability (class in pytorch_utils.modules)

N

nb_distinct_values (CategoricalFeatureEmbedding attribute) NotMonotone

NotNonDecreasing NotNonIncreasing

O

output_categorical_features (AugmentedBernoulliDataModule property) output_features (AugmentedBernoulliDataModule property)

output_features_dtypes (AugmentedBernoulliDataModule property) output_real_features (AugmentedBernoulliDataModule property)

P

pandas_formatter (AugmentedBernoulliDatasetConfig attribute) Partitioned (class in pytorch_utils.modules) plot_probability_mapping() (MonotoneBernoulliProbability method) predict (MLStage attribute) predict() (MonotoneBernoulliProbability method) predict_dataloader() (AugmentedBernoulliDataModule method) predict_from_pandas() (MonotoneBernoulliProbability method) (ProbabilityPredictor method) predict_step() (MonotoneBernoulliProbability method) prepare_data() (AugmentedBernoulliDataModule method) preprocessing_pandas_collate_fn() (AugmentedBernoulliDataModule method) probability_mapping() (MonotoneBernoulliProbability method) ProbabilityPredictor (class in pytorch_utils.modules) pytorch_utils module

pytorch_utils.data_modules module pytorch_utils.dataset_configurations module pytorch_utils.datasets module pytorch_utils.exceptions module pytorch_utils.metrics module pytorch_utils.miscellaneous module pytorch_utils.modules module pytorch_utils.utils module

R

random_seed (DataSplitConfig attribute), [1]

raw_feature_names (AugmentedBernoulliDataset property)

S

sample() (AugmentedBernoulliDatasetConfig method) (AugmentedBernoulliDatasetConfigs method) sample_weight_col (AugmentedBernoulliDataset attribute) scale_col() (DataAugmentationConfig static method) scaling_factors (DataAugmentationConfig attribute), [1] scaling_filter() (DataAugmentationConfig static method)

scaling_length() (DataAugmentationConfig static method) setup() (AugmentedBernoulliDataModule method) setup_datasets() (AugmentedBernoulliDataModule method) ShiftedEmbedding (class in pytorch_utils.modules) sort_hashable() (in module pytorch_utils.miscellaneous) split_config (AugmentedBernoulliDatasetConfig attribute), [1], [2] stratify (DataSplitConfig attribute), [1]

T

TaskNotFoundError test (MLStage attribute) test_data (AugmentedBernoulliDatasetConfig property) test_data_length (AugmentedBernoulliDatasetConfig property) test_dataloader() (AugmentedBernoulliDataModule method) test_proportion (DataSplitConfig attribute), [1] test_step() (MonotoneBernoulliProbability method) train_dataloader() (AugmentedBernoulliDataModule method)

train_valid_test_split() (DataSplitConfig method) training_data (AugmentedBernoulliDatasetConfig property) training_data_length (AugmentedBernoulliDatasetConfig property) training_proportion (DataSplitConfig attribute), [1] training_step() (MonotoneBernoulliProbability method) transform() (AugmentedBernoulliDataModule method) transform_to_tensors() (AugmentedBernoulliDataModule method) transformed_feature_names (AugmentedBernoulliDataset property)

U

update() (WeightedBinaryCalibrationError method) (WeightedMeanAbsoluteError method) (WeightedMeanSquaredError method)

V

val_dataloader() (AugmentedBernoulliDataModule method) validate (MLStage attribute) validation_data (AugmentedBernoulliDatasetConfig property)

validation_data_length (AugmentedBernoulliDatasetConfig property) validation_proportion (DataSplitConfig attribute), [1] validation_step() (MonotoneBernoulliProbability method)

W

WeightedBinaryCalibrationError (class in pytorch_utils.metrics)

WeightedMeanAbsoluteError (class in pytorch_utils.metrics) WeightedMeanSquaredError (class in pytorch_utils.metrics)