Dataset

class niaarm.dataset.Dataset(path_or_df, delimiter=',', header=0, names=None)

Bases: object

Class for working with a dataset.

Parameters:
  • path_or_df (Union[str, os.PathLike, pandas.DataFrame]) – Path to the dataset (csv) file or a pandas DataFrame.

  • delimiter (str) – The delimiter in the csv file.

  • header (Optional[int]) – Row to use as header (zero-based). Default: 0. Pass header=None if the file doesn’t contain a header.

  • names (Optional[list[str]]) – List of feature names to use. If the file already contains a header row, pass header=0 to override the feature names.

transactions

Transactional data.

Type:

pandas.DataFrame

header

Feature names.

Type:

list[str]

features

List of features.

Type:

list[Feature]

dimension

Dimension of the optimization problem for the dataset.

Type:

int