Skip to content

Commit 0828f01

Browse files
authored
Add aliases for renamed spektral layers. Fixes #239 (#240)
* Add aliases for renamed spektral layers. Fixes #239 * Update keras seed
1 parent 682a900 commit 0828f01

File tree

2 files changed

+18
-0
lines changed

2 files changed

+18
-0
lines changed

src/plugins/CreateKerasMeta/schemas/index.js

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,8 +93,26 @@ define([
9393
returnMask.default = 'True';
9494
}
9595

96+
addKnownAliases(layer);
97+
9698
return layer;
9799
});
98100

101+
function addKnownAliases(layer) {
102+
const KNOWN_LAYER_ALIASES = {
103+
GCNConv: 'GraphConv',
104+
ECCConv: 'EdgeConditionedConv',
105+
GATConv: 'GraphAttention',
106+
GCSConv: 'GraphConvSkip',
107+
APPNPConv: 'APPNP'
108+
};
109+
110+
const alias = KNOWN_LAYER_ALIASES[layer.name];
111+
if (alias) {
112+
layer.aliases.push(alias);
113+
}
114+
return layer;
115+
}
116+
99117
return {SpecialTypes, Layers};
100118
});

src/seeds/keras/keras.webgmex

3.06 KB
Binary file not shown.

0 commit comments

Comments
 (0)