-
Notifications
You must be signed in to change notification settings - Fork 1
Open
Labels
enhancementNew feature or requestNew feature or request
Description
Graph-based proxies are all the rage recently, but they can be a bit cumbersome to implement. We can alleviate that in L2O by extracting a graph representation automatically so users can easily build a GNN model. I imagine an API like:
import LearningToOptimize as L2O
import GraphNeuralNetworks as GNN
model::JuMP.Model = ...
encoder = L2O.graph_encoder(L2O.Bipartite, GNN.GATConv, model; out=128)
gnn = Flux.Chain(
encoder,
Dense(128,1)
)
https://github.com/lanl-ansi/MathProgIncidence.jl should be very helpful for this.
andrewrosemberg
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request