synthval.utilities.FeaturesDataset#

class synthval.utilities.FeaturesDataset(ori_features, gen_features, test_percentage=0.2, rng_seed=0, train_mode=True)#

Bases: pynever.datasets.Dataset

Utility class for the management of the dataset composed by the feature extracted from the original images and the synthetically generated images. Needed for using the pynever training and testing strategies.

Parameters:

  • ori_features (pandas.DataFrame)

  • gen_features (pandas.DataFrame)

  • test_percentage (float)

  • rng_seed (int)

  • train_mode (bool)

train_df#

The training set.

Type:

pandas.Dataframe

test_df#

The test set.

Type:

pandas.Dataframe

train_mode#

Flag controlling if the __get_item__ function returns element from the training set or the test set.

Type:

bool