core.data

core.data

Classes

Name	Description
SpotDataFromArray	Data handler for numpy arrays or torch tensors.
SpotDataFromTorchDataset	Data handler for PyTorch Datasets.
SpotDataSet	Abstract base class for data handling in SpotOptim.

SpotDataFromArray

core.data.SpotDataFromArray(
    x_train,
    y_train,
    x_val=None,
    y_val=None,
    x_test=None,
    y_test=None,
    target_column=None,
)

Data handler for numpy arrays or torch tensors.

SpotDataFromTorchDataset

core.data.SpotDataFromTorchDataset(
    train_dataset,
    input_dim,
    output_dim,
    val_dataset=None,
    test_dataset=None,
    target_column=None,
)

Data handler for PyTorch Datasets.

SpotDataSet

core.data.SpotDataSet(input_dim, output_dim, target_column=None)

Abstract base class for data handling in SpotOptim.

Attributes

Name	Type	Description
input_dim	int	Number of input features.
output_dim	int	Number of output features.
target_column	str	Name of the target column if applicable.

Methods

Name	Description
get_test_data	Returns the test data.
get_train_data	Returns the training data.
get_validation_data	Returns the validation data.

get_test_data

core.data.SpotDataSet.get_test_data()

Returns the test data.

get_train_data

core.data.SpotDataSet.get_train_data()

Returns the training data.

get_validation_data

core.data.SpotDataSet.get_validation_data()

Returns the validation data.