def append_train_args(base_args, data_name, path, uncertainty):
    """
    Appends dataset and UQ specific args to the base train args for training.
    :dict base_args: basic training args
    :str data_name: name of dataset currently being used
    :str path: path of dataset currently being used
    :str uncertainty: UQ method being used
    """

    save_dir = 'ckpnts/' + uncertainty + '_' + data_name[:-4]
    features_path = 'features/' + data_name[:-4] + '_features.npz'

    additional_args = {'--data_path': path,
                       '--save_dir': save_dir,
                       '--uncertainty': uncertainty,
                       '--features_path': features_path}
    base_copy = base_args.copy()
    base_copy.update(additional_args)

    return base_copy