Реализация обработки набора данных для потери триплета со многими классами
В настоящее время я работаю над внедрением обучения с потерей триплета в тензорном потоке. Моя проблема в том, что API tf.data.Dataset для меня является своего рода черным ящиком, однако я хочу создать свой набор данных, а затем загрузить ~20 различных классов и 5 представителей для каждого класса в каждом пакете. Знаете ли вы какой-либо способ создания пользовательской пакетной функции? Мне пришла в голову идея сначала загрузить 20 якорей, а для каждого якоря 5 позитивов из этого проекта: https://www.sentiance.com/2018/05/03/venue-mapping/. Но я не знаю, как можно реализовать эту идею в тензорном потоке.