fix

kumo-ai · Nov 17, 2024 · de01e3e · de01e3e
1 parent 5ba7bc4
commit de01e3e
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 3 deletions.
diff --git a/convergence_curve_contextgnn.png b/convergence_curve_contextgnn.png
diff --git a/convergence_curve_idgnn.png b/convergence_curve_idgnn.png
diff --git a/convergence_curve_shallowrhsgnn.png b/convergence_curve_shallowrhsgnn.png
diff --git a/examples/.relbench_example.py.swp b/examples/.relbench_example.py.swp
diff --git a/examples/contextgnn_sample_softmax.py b/examples/contextgnn_sample_softmax.py
@@ -1,7 +1,7 @@
-"""Example script to run the models in this repository.
+"""Example script to run sample softmax on ContextGNN.
 
-python relbench_example.py --dataset rel-trial --task site-sponsor-run
-    --model contextgnn --epochs 10
+python contextgnn_sample_softmax.py --dataset rel-trial --task site-sponsor-run
+    --epochs 10
 """
 
 import argparse