kumoai.trainer.GNNModelPlan#
- class kumoai.trainer.GNNModelPlan[source]#
Configuration parameters that define how the Kumo graph neural network is architected.
- Variables:
aggregation – (
list[list["sum" | "mean" | "min" | "max" | "std"]]) A nested list of aggregation operators in the Graph Neural Network aggregation process (default:[ ["sum", "mean", "max"], ["sum", "mean", "min", "max", "std"]]). Supported Task Types: All