import torch
import numpy as numpy

class ExperimentBuilder(object):
    def __init__(self,data):
        self.data = data


    def run_experiment():
        ## training and validate
        for epoch ...
            ## TODO run train
            self.data.get_train_batches

            ## TODO run validate

        ##test

