tf.lookup.TextFileInitializer

Инициализаторы таблицы из текстового файла.

tf.lookup.TextFileInitializer(
    filename, key_dtype, key_index, value_dtype, value_index, vocab_size=None,
    delimiter='\t', name=None
)

Этот инициализатор назначает одну запись в таблицу для каждой строки в файле.

Тип ключа и значения таблицы для инициализации задаётся key_dtype и value_dtype.

Содержание ключа и значения, которые нужно получить из каждой строки, задаётся key_index и value_index.

• TextFileIndex.LINE_NUMBER означает использование номера строки, начиная с нуля, ожидается тип данных int64.
• TextFileIndex.WHOLE_LINE означает использование всего содержимого строки, ожидается тип данных строка.
• Значение >=0 означает использование индекса (начиная с нуля) разделённой строки на основе delimiter.

Например, если у нас есть файл со следующим содержимым:

emerson 10
lake 20
palmer 30

Следующий фрагмент кода инициализирует таблицу с первым столбцом в качестве ключей и вторым столбцом в качестве значений:

• emerson -> 10
• lake -> 20
• palmer -> 30

table = tf.lookup.StaticHashTable(tf.lookup.TextFileInitializer(
    "test.txt", tf.string, 0, tf.int64, 1, delimiter=" "), -1)
...
table.init.run()

Аналогично, для инициализации всей строки в качестве ключей и номера строки в качестве значений.

• emerson 10 -> 0
• lake 20 -> 1
• palmer 30 -> 2

table = tf.lookup.StaticHashTable(tf.lookup.TextFileInitializer(
    "test.txt", tf.string, tf.lookup.TextFileIndex.WHOLE_LINE,
    tf.int64, tf.lookup.TextFileIndex.LINE_NUMBER, delimiter=" "), -1)
...
table.init.run()

Методы

initialize

Просмотреть исходный код

initialize(
    table
)

Инициализирует таблицу из текстового файла.