Блог

Я использую функцию make_csv_dataset для передачи данных для обучения модели. Поскольку я использую tf.keras.Модель.функция подгонки, мне нужно передать (входные данные, цели, sample_weights) кортеж для функции подгонки из объекта tf.data. Как я могу использовать функцию make_csv_dataset для генерации этого кортежа?(У меня есть столбец weight и столбец label в моем csv рядом с другими столбцами функций)

В настоящее время в make_csv_dataset есть только параметр label_name для указания столбца label, но как я могу указать столбец weight для этого объекта tf.data?

Для этого вопроса приемлемо любое решение, заранее спасибо!