MilesCranmer commited on
Commit
1ec3ee8
·
unverified ·
1 Parent(s): 0d22412

Reduce length of DL part of tutorial

Browse files
Files changed (1) hide show
  1. 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 = 50000\n",
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=3000)\n",
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",