def make_tfset(train_names, val_names, name):
tfset_pb = pb.TfSet()
tfset_pb.train[:] = train_names
tfset_pb.val[:] = val_names
tfset_pb.name = name
return tfset_pb
def read_tfset(path):
data_pb = pb.TfSet()
read_from_text_file(path, data_pb)
return data_pb