¿Cómo establecer class_weight en el paquete keras de R?
estoy usandokeras
paquete en R para entrenar un modelo de aprendizaje profundo. Mi conjunto de datos está altamente desequilibrado. Por lo tanto, quiero establecerclass_weight
argumento en elfit
función. Aquí está la función de ajuste y sus argumentos que utilicé para mi modelo
history <- model %>% fit(
trainData, trainClass,
epochs = 5, batch_size = 1000,
class_weight = ????,
validation_split = 0.2
)
En python puedo configurarclass_weight
de la siguiente manera:
class_weight={0:1, 1:30}
Pero no estoy seguro de cómo hacerlo en R. En el menú de ayuda de R se describeclass_weight
de la siguiente manera:
Índices de asignación de listas con nombre (enteros) opcionales a un peso (flotante) para aplicar a la pérdida del modelo para las muestras de esta clase durante el entrenamiento. Esto puede ser útil para decirle al modelo que "preste más atención" a las muestras de una clase subrepresentada.
¿Alguna idea o sugerencia?