Commit 6e4c9721 authored by JGlombitza's avatar JGlombitza
Browse files

add testfile

parent cabcc01a
from tensorflow import keras
from tensorflow.keras import layers as lay
from edgeconv import EdgeConv
def f(x):
x = lay.Dense(30)(x)
x = lay.Dense(20)(x)
return x
inp = lay.Input((10,))
a = EdgeConv(f, next_neighbors=3)(inp)
y = EdgeConv(f, next_neighbors=3)(a)
out = EdgeConv(f, next_neighbors=3)(y)
model = keras.models.Model(inp, out)
model.summary()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment