batch_data_read.py

def get_train_data_batch(n):
    features = []
    with open('data.csv') as f:
        count=0
        for line in f:
            splitted = line.split(';')
            featureset = [int(float(x)) for x in splitted[:len(splitted)-1]]
            label = int(float(splitted[-1].rstrip()))
            count+=1
            if count % n==0:
                features = np.array(features)
                train_x = list(features[:,0])
                train_y = list(features[:,1])
                yield train_x, train_y
                features = []