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