Мое требование состоит в том, чтобы классифицировать изображения по 7 заранее определенным категориям (Люди, Животные, Вечеринки / События, Природа, Еда, Спорт, Искусство)
Я использовал изображения, относящиеся к каждой категории, и отметил их50 слов каждый.(изображения людей, помеченных словами, относящимися к картинке и т. д.) Теги, созданные с помощью API.
Теперь у меня есть список тегов для каждого изображения вместе с категорией, к которой оно относится.
Теперь я хочу обучить модель идентифицировать слова, принадлежащие каждой категории, и определить категорию изображения, если даны теги этого изображения.
Есть идеи, как сделать это наиболее эффективно?
Я пробовал LDA, но не думаю, что это соответствует моим целям, так как я хочу обучить модель вместе с тегами и категориями и прогнозировать категорию на основе другого набора тегов.