Добавьте sample_weight/class_weight при использовании GCN
Я только начал изучать, как классифицировать узлы с помощью сверточных сетей Graph (GCN), следуя этому примеру из официальной документации StellarGraph (https://stellargraph.readthedocs.io/en/v1.0.0rc1/demos/node-classification/gcn /gcn-cora-node-classification-example.html ) в Cora Dataset, а архитектура модели показана на изображении ниже.
Но я пытаюсь добавить в модель class_weight или sample_weight , потому что данные несбалансированы. Во-первых, когда я пытаюсь добавить class_weight, я получаю следующую ошибку:
ValueError: class_weight не поддерживается для трехмерных целей.
И при попытке использовать sample_weight я все еще получал ошибку:
ValueError: аргумент sample_weight не поддерживается при использовании keras.utils.Sequence в качестве входных данных.
и, как написано в официальной документации, причина в том, что я использую генератор: "Этот аргумент не поддерживается, когда x является набором данных, генератором или экземпляром keras.utils.Sequence, вместо этого укажите sample_weights в качестве третьего элемента x . " (и я использую FullBatchNodeGenerator)
Кроме того, я попытался сгладить слой над плотным слоем, но все равно безуспешно, поскольку flatten (1, None, 8) приводит к (1, None), что недопустимо для ввода в Dense слой (я получил
The last dimension of the inputs to Dense should be defined. Found "None". ошибка).
Может быть, у кого-то из вас возникла та же проблема, что и у новичка, и вы хотели бы помочь мне найти способ ее решить?
Спасибо.