Fixed directory handling - may still be temperamental

from sys import stdout

def model_train_test(data, model, device, n_epochs, n_batches, loss_function, learning_rate, verbose=False):

def model_train_test(data, model, device, n_epochs, n_batches, loss_function, learning_rate, verbose=False, return_losses=False):

    xtrain, ytrain, xtest, ytest = data

    name = model.name

    #plt.show()

    plt.close()

    out = (model,)

    if return_losses:

        out += (train_losses, test_losses,)

    return model

def get_script_path():

    return os.path.dirname(os.path.realpath(sys.argv[0]))

    #return os.path.dirname(os.path.realpath(sys.argv[0]))

    return os.getcwd()

def norm(dataframe, ref_dataframe=None, ref_mean=None, ref_std=None):