DOC smooth speed plot

Author: MatthewSZhang

.github/workflows/emscripten.yml

@@ -9,7 +9,7 @@ jobs:
     steps:
       - uses: actions/checkout@v4
       - name: Build WASM wheel
-        uses: pypa/[email protected].2
+        uses: pypa/[email protected].3
         env:
           CIBW_PLATFORM: pyodide
       - name: Upload package

.github/workflows/wheel.yml

@@ -33,7 +33,7 @@ jobs:
     steps:
       - uses: actions/checkout@v4
       - name: Build wheels
-        uses: pypa/[email protected].2
+        uses: pypa/[email protected].3
         env:
           CIBW_SKIP: "*_i686 *_ppc64le *_s390x *_universal2 *-musllinux_* cp314*"
           CIBW_PROJECT_REQUIRES_PYTHON: ">=3.10"

examples/plot_speed.py

@@ -174,19 +174,25 @@ def baseline(X, y, t):
 
 n_features_max = 30
 
+from timeit import repeat
+
 time_h = np.zeros(n_features_max, dtype=float)
 time_eta = np.zeros(n_features_max, dtype=float)
 for i in range(n_features_max):
-    time_h[i] = timeit(
+    times_h = repeat(
         f"s = FastCan({i + 1}, verbose=0).fit(X, y)",
-        number=10,
+        number=1,
+        repeat=10,
         globals=globals(),
     )
-    time_eta[i] = timeit(
+    time_h[i] = np.median(times_h)
+    times_eta = repeat(
         f"s = FastCan({i + 1}, eta=True, verbose=0).fit(X, y)",
-        number=10,
+        number=1,
+        repeat=10,
         globals=globals(),
     )
+    time_eta[i] = np.median(times_eta)
 
 feature_num = np.arange(n_features_max, dtype=int) + 1
 plt.plot(feature_num, time_h, label="h-correlation")