def get_data():
    csv_file_path = '../data/the_data_file.csv'

    if os.path.exists(csv_file_path):
        print("Data file found, loading data...")
        with open(csv_file_path, "r") as f:
            data = pd.read_csv(f)

    else:
        print("Data file not found, assembling dataset...")
        from data import join_data as data
        data, labels = data()
        data.to_csv(csv_file_path, index=False)

    return data
Exemplo n.º 2
0
        if result == 'plus10':
            result = 1.2
        elif result == 'minus10':
            result = 0.8
        else:
            result = 1
        results_nums.append(result)
    X['uninsured'] *= results_nums[0]
    X['unemployment'] *= results_nums[1]
    X['obese_adult'] *= results_nums[2]
    X['smoke_adult'] *= results_nums[3]
    X['air_poll_partic'] *= results_nums[4]

    return X


if __name__ == '__main__':

    csv_file_path = '../data/the_data_file.csv'
    if os.path.exists(csv_file_path):
        print("Data file found, loading data...")
        with open(csv_file_path, "r") as f:
            data = pd.read_csv(f)
    else:
        print("Data file not found, assembling dataset...")
        # from combine_data import join_data as data
        from data import join_data as data
        # data, labels = data()
        data = data()
        data.to_csv(csv_file_path, index=False)