- class DatasetFromDataframe(*args, **kwargs)[source]
Bases:
torch.utils.data.Dataset- Attributes:
num_samples
Returns the number of samples in the dataset.
Methods
__call__(*args, **kwargs)Call self as a function.
convert_to_validation(model)Converts the dataset to validation mode by resetting instance variables.
get_train_dataset(model, df)A helper function to get a train dataset with the provided parameters.
get_validation_dataset(model, df)A helper function to get a validation dataset with the provided parameters.
- convert_to_validation(model)[source]
Converts the dataset to validation mode by resetting instance variables.
- Parameters:
- modelAutoEncoder
The autoencoder model used to get relevant params and the preprocessing func.
- static get_train_dataset(model, df)[source]
A helper function to get a train dataset with the provided parameters.
- Parameters:
- modelAutoEncoder
- dfpandas.DataFrame
The autoencoder model used to get relevant params and the preprocessing func.
Input dataframe used for the dataset.
- Returns:
- DatasetFromDataframe
Training Dataset set up to load from the dataframe.
- static get_validation_dataset(model, df)[source]
A helper function to get a validation dataset with the provided parameters.
- Parameters:
- modelAutoEncoder
- dfpandas.DataFrame
The autoencoder model used to get relevant params and the preprocessing func.
Input dataframe used for the dataset.
- Returns:
- DatasetFromDataframe
Validation Dataset set up to load from the dataframe.
- property num_samples
Returns the number of samples in the dataset.
