Skip to content
Snippets Groups Projects
Dataset.py 565 B
Newer Older
import pandas as pd


class Dataset:
    def __init__(self, dataset_filename: str) -> None:
        raw_dataset = pd.read_csv(dataset_filename, encoding="windows-1254")
        self.dataset = self.preprocess_dataset(raw_dataset)

    def preprocess_dataset(self, raw_dataframe):
        # preprocessing goes here
        return raw_dataframe

    def draw_histogram(self):
        raise NotImplementedError

    def get_dataset_columns(self) -> list:
        raise NotImplementedError

    def get_plottable_columns(self) -> list:
        raise NotImplementedError