cgnal.core.datasets package

Module contents

Module containing samples datasets to be used for testing.

cgnal.core.datasets.get_unbalanced_dataset(features_file: str = '/home/runner/work/cgnal-core/cgnal-core/cgnal/core/datasets/data/unbalanced_features.p', labels_file: str = '/home/runner/work/cgnal-core/cgnal-core/cgnal/core/datasets/data/unbalanced_labels.p') cgnal.core.data.model.ml.PandasDataset

Return unbalanced dataset.

Parameters
  • features_file – name of the pickle file where the features data are stored.

  • labels_file – name of the pickle file where the labels data are stored.

Returns

PandasDataset object with unbalanced classes.

cgnal.core.datasets.get_weather_nyc_dataset(filename: str = '/home/runner/work/cgnal-core/cgnal-core/cgnal/core/datasets/data/weather_nyc_short.csv') cgnal.core.data.model.ml.PandasTimeIndexedDataset

Return the NYC Wheather dataset.

Parameters

filename – name of the csv file where the Wheather NYC data are stored.

Returns

PandasTimeIndexedDataset object with the NYC Wheather dataset.