Closed CarloLucibello closed 3 years ago
It seems that there is not much need for FeatureSelector, since
FeatureSelector
feature = :node model = Flux.Chain( GCNConv(1024=>256, relu), FeatureSelector(feature), softmax )
can be easily done with
model = Flux.Chain( GCNConv(1024=>256, relu), node_feature, # or edge_feature or global_feature softmax )
More complex behavior can be implemente with closures, e.g.
model = Flux.Chain( GCNConv(1024=>256, relu), fg -> (node_feature(fg), edge_feature(fg)) (x, e) -> (softmax(x), sigmoid.(e)) )
I agree with using node_feature, since I saw someone else used it before. It is worth documentation.
node_feature
It seems that there is not much need for
FeatureSelector
, sincecan be easily done with
More complex behavior can be implemente with closures, e.g.