Skip to content

Commit 21813ae

Browse files
committed
Fixed codebeat error
1 parent b68eaaf commit 21813ae

6 files changed

+4
-1
lines changed

src/main.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,10 @@ def test(model, x):
4545
for graph_pair in tqdm(test):
4646
data = process(graph_pair)
4747
data = convert_to_keras(data, global_labels)
48-
x, y, a, b = [ np.array([ data["features_1"] ]), np.array([ data["features_2"] ]), np.array([ data["edge_index_1"] ]), np.array([ data["edge_index_2"] ]) ]
48+
x = np.array([ data["features_1"] ])
49+
y = np.array([ data["features_2"] ])
50+
a = np.array([ data["edge_index_1"] ])
51+
b = np.array([ data["edge_index_2"] ])
4952
g_truth.append(data["target"])
5053
y=model.predict([x, a, y, b])
5154
scores.append(find_loss(y, data["target"]))

train/saved_model.pb

-35 KB
Binary file not shown.
78 KB
Binary file not shown.

train/variables/variables.index

3.02 KB
Binary file not shown.

xweights.data-00000-of-00001

0 Bytes
Binary file not shown.

xweights.index

0 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)