loading.py

import tensorflow as tf

def load_dataset(filename, label):
    # Loading the dataset file
    dataset = tf.data.experimental.CsvDataset(filename,   [tf.float32, ], select_cols=[2], header=True, field_delim=" ")

    # Reshape the incoming data
    dataset = dataset.batch(200)

    # Create labelled vector
    dataset = dataset.map(lambda x: (tf.reshape(x,(200,1)),label))

    return dataset


def combine_datasets(datasets):
    # Combining the datasets

    c_dataset = datasets[0].concatenate(datasets[1])

    for dataset in datasets[2:]:
        c_dataset = c_dataset.concatenate(dataset)

    return c_dataset
    
def split_train_test_validation(dataset, train_size, test_size, val_size):
    train_dataset = dataset.take(train_size)
    test_dataset = dataset.take(test_size)
    val_dataset = dataset.take(val_size)