Spaces:
Sleeping
Sleeping
Reduce length of DL part of tutorial
Browse files- examples/pysr_demo.ipynb +2 -2
examples/pysr_demo.ipynb
CHANGED
@@ -1064,7 +1064,7 @@
|
|
1064 |
"outputs": [],
|
1065 |
"source": [
|
1066 |
"hidden = 128\n",
|
1067 |
-
"total_steps =
|
1068 |
"\n",
|
1069 |
"\n",
|
1070 |
"def mlp(size_in, size_out, act=nn.ReLU):\n",
|
@@ -1284,7 +1284,7 @@
|
|
1284 |
"np.random.seed(1)\n",
|
1285 |
"tmpX = X_for_pysr.detach().numpy().reshape(-1, 5)\n",
|
1286 |
"tmpy = y_i_for_pysr.detach().numpy().reshape(-1)\n",
|
1287 |
-
"idx2 = np.random.randint(0, tmpy.shape[0], size=
|
1288 |
"\n",
|
1289 |
"model = PySRRegressor(\n",
|
1290 |
" niterations=20,\n",
|
|
|
1064 |
"outputs": [],
|
1065 |
"source": [
|
1066 |
"hidden = 128\n",
|
1067 |
+
"total_steps = 10_000\n",
|
1068 |
"\n",
|
1069 |
"\n",
|
1070 |
"def mlp(size_in, size_out, act=nn.ReLU):\n",
|
|
|
1284 |
"np.random.seed(1)\n",
|
1285 |
"tmpX = X_for_pysr.detach().numpy().reshape(-1, 5)\n",
|
1286 |
"tmpy = y_i_for_pysr.detach().numpy().reshape(-1)\n",
|
1287 |
+
"idx2 = np.random.randint(0, tmpy.shape[0], size=500)\n",
|
1288 |
"\n",
|
1289 |
"model = PySRRegressor(\n",
|
1290 |
" niterations=20,\n",
|