Добавьте sample_weight/class_weight при использовании GCN

Question

Я только начал изучать, как классифицировать узлы с помощью сверточных сетей Graph (GCN), следуя этому примеру в официальной документации StellarGraph ( ) в https://stellargraph.readthedocs.io/en/v1.0.0rc1/demos/node-classification/gcn/gcn-cora-node-classification-example.htmlнаборе данных Cora.

Но я пытаюсь добавить в модель class_weight или sample_weight, потому что данные несбалансированы. Во-первых, когда я пытаюсь добавить class_weight, я получаю следующую ошибку:

ValueError: class_weight не поддерживается для трехмерных целей.

И при попытке использовать sample_weight я все еще получал ошибку:

ValueError: sample_weight аргумент не поддерживается при использовании keras.utils.Sequence как вход.

и причина, как написано в официальной документации, "This argument is not supported when x is a dataset, generator, or keras.utils.Sequence instance, instead provide the sample_weights as the third element of x." (и я использую FullBatchNodeGenerator)

Может быть, у кого-то из вас возникла та же проблема, что и у новичка, и помогите мне узнать, как ее решить?

Спасибо!