INNER CODE UNIT · Python

load_data

ECSIM/pem-dataset1 · Notebooks/functions.py:26

def load_data(path, set_flag=False):
    """
    Load dataset.

    :param path: file path
    :type path: str
    :param set_flag: activation set flag
    :type set_flag: bool
    :return: data as numpy array
    """
    file = open(path, "r")
    data = []
    set_list = []
    lines = file.readlines()[1:]
    for line in lines:
        splitted_line = line.split(",")
        if not set_flag:
            data.append(list(map(float, splitted_line)))

View source record →