aaaa

examples/test_all.sh

@@ -41,7 +41,10 @@ mkdir -p all_"$seed"
 for data in "${data_param[@]}"; do
     for topk in "${topk_list[@]}"; do
         model="TGN_large"
-        
+        probability_params=("0" "0.0001" "0.001" "0.01" "0.1" "1")
+        if [ "$data" = "StackOverflow" ]; then
+            probability_params=("0.0001" "0.001" )
+        fi
         neighbor="20"
         if [ "$data" = "WIKI" ] || [ "$data" = "REDDIT" ] || [ "$data" = "LASTFM" ]; then
             model="TGN"