Метка классификации Tensorflow с кодировкой 1000 классов

Question

У меня есть проблема классификации нескольких классов в Tensorflow. Метка является строковым типом и имеет 1000 уникальных значений. Как это закодировать?

если я передаю его прямо как метку, получаю эту ошибку

ValueError: Labels dtype should be integer Instead got <dtype: 'string'>.

Answer 1

Вы не предоставили никакого кода, поэтому я не знаю, где вы на самом деле передаете ярлыки. Но я могу дать общий ответ.

В задаче классификации с известным числом классов вы просто назначаете целое число каждому классу. Так что в вашем случае вы можете создать словарь python, который бы отображал ваши слова в целое число, например:

word_to_index = {'word1': 0, 'word2': 1, 'word3': 2}
label = 'word2'
index = word_to_index[label]

От того, как вы поставили свой вопрос и ошибку, которую вы получили (говорит integer), мне кажется, что API, который вы используете, просто ожидает такое целое число.