Commit f874f631 authored by Niklas Uwe Langner's avatar Niklas Uwe Langner
Browse files

[test] add compiling and training to test model

parent 00648d07
from tensorflow import keras
from tensorflow.keras import layers as lay
from edgeconv import EdgeConv
import numpy as np
def f(x):
......@@ -17,3 +18,7 @@ out = EdgeConv(f, next_neighbors=3)(y)
model = keras.models.Model([points, feats], out)
model.summary()
model.compile(loss="mse", optimizer=keras.optimizers.Adam())
model.fit([np.ones((16, 10, 5)), np.ones((16, 10, 5))], np.ones((16, 10, 20)), epochs=10)
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