Dividir el tensor en conjuntos de entrenamiento y prueba
Digamos que he leído en un archivo de texto usando unTextLineReader
. ¿Hay alguna manera de dividir esto en trenes y conjuntos de prueba enTensorflow
? Algo como:
def read_my_file_format(filename_queue):
reader = tf.TextLineReader()
key, record_string = reader.read(filename_queue)
raw_features, label = tf.decode_csv(record_string)
features = some_processing(raw_features)
features_train, labels_train, features_test, labels_test = tf.train_split(features,
labels,
frac=.1)
return features_train, labels_train, features_test, labels_test