Como definir class_weight no pacote keras de R?
estou usandokeras
pacote em R para treinar um modelo de aprendizado profundo. Meu conjunto de dados está altamente desequilibrado. Portanto, eu quero definirclass_weight
argumento nofit
função. Aqui está a função de ajuste e seus argumentos que eu usei para o meu modelo
history <- model %>% fit(
trainData, trainClass,
epochs = 5, batch_size = 1000,
class_weight = ????,
validation_split = 0.2
)
Em python eu posso definirclass_weight
como se segue:
class_weight={0:1, 1:30}
Mas não tenho certeza de como fazê-lo em R. No menu de ajuda do R, ele descreveclass_weight
como se segue:
Índices opcionais de mapeamento de lista nomeada (números inteiros) para um peso (flutuante) para aplicar à perda do modelo para as amostras dessa classe durante o treinamento. Isso pode ser útil dizer ao modelo para "prestar mais atenção" às amostras de uma classe sub-representada.
Alguma ideia ou sugestão?