Skip to content

Commit

Permalink
[Modify] Fixed some bug
Browse files Browse the repository at this point in the history
  • Loading branch information
Y-nuclear committed Sep 23, 2024
1 parent cc4b99a commit 2a1657b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/gnnwr/datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -1003,7 +1003,7 @@ def _init_gtnnwr_stpnn_distance_pred(

# calculate point matrix
matrix_length = len(refer_s_point)
pred_length,val_length,test_length = len(pred_s_point),len(val_s_point),len(test_s_point)
pred_length = len(pred_s_point)
pred_s_point_matrix = np.repeat(pred_s_point[:, np.newaxis, :], matrix_length, axis=1)
refer_s_point_pred_matrix = np.repeat(refer_s_point[:,np.newaxis,:], pred_length, axis=1)
pred_s_point_matrix = np.concatenate(
Expand Down

0 comments on commit 2a1657b

Please sign in to comment.