Fix typo in user guide.

stanfordmlgroup · Jun 4, 2020 · c83672f · c83672f
1 parent 5acfefe
commit c83672f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/docs/1-useage.html b/docs/1-useage.html
@@ -26,8 +26,8 @@
   "@type": "NewsArticle",
   "mainEntityOfPage": "https://stanfordmlgroup.github.io/ngboost/1-useage.html",
   "headline": "Usage",
-  "datePublished": "2020-03-31T11:59:13-07:00",
-  "dateModified": "2020-03-31T11:59:13-07:00",
+  "datePublished": "2020-06-03T18:28:35-07:00",
+  "dateModified": "2020-06-03T18:28:35-07:00",
   "description": "        Usage    We'll start with a probabilistic regression example on the Boston housing dataset:    from ngboost import NGBRegressorfrom sklearn.datasets ...",
   "author": {
     "@type": "Person",
@@ -1086,8 +1086,8 @@ <h4 id="Survival-Regression">Survival Regression<a class="anchor-link" href="#Su
 <span class="n">X_surv_train</span><span class="p">,</span> <span class="n">X_surv_test</span><span class="p">,</span> <span class="n">Y_surv_train</span><span class="p">,</span> <span class="n">Y_surv_test</span> <span class="o">=</span> <span class="n">train_test_split</span><span class="p">(</span><span class="n">X</span><span class="p">,</span> <span class="n">Y</span><span class="p">,</span> <span class="n">test_size</span><span class="o">=</span><span class="mf">0.2</span><span class="p">)</span>
 
 <span class="c1"># introduce administrative censoring to simulate survival data</span>
-<span class="n">T_surv_train</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">minimum</span><span class="p">(</span><span class="n">Y_train</span><span class="p">,</span> <span class="mi">30</span><span class="p">)</span> <span class="c1"># time of an event or censoring</span>
-<span class="n">E_surv_train</span> <span class="o">=</span> <span class="n">Y_train</span> <span class="o">&gt;</span> <span class="mi">30</span> <span class="c1"># 1 if T[i] is the time of an event, 0 if it&#39;s a time of censoring</span>
+<span class="n">T_surv_train</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">minimum</span><span class="p">(</span><span class="n">Y_surv_train</span><span class="p">,</span> <span class="mi">30</span><span class="p">)</span> <span class="c1"># time of an event or censoring</span>
+<span class="n">E_surv_train</span> <span class="o">=</span> <span class="n">Y_surv_train</span> <span class="o">&gt;</span> <span class="mi">30</span> <span class="c1"># 1 if T[i] is the time of an event, 0 if it&#39;s a time of censoring</span>
 
 <span class="n">ngb</span> <span class="o">=</span> <span class="n">NGBSurvival</span><span class="p">(</span><span class="n">Dist</span><span class="o">=</span><span class="n">LogNormal</span><span class="p">)</span><span class="o">.</span><span class="n">fit</span><span class="p">(</span><span class="n">X_surv_train</span><span class="p">,</span> <span class="n">T_surv_train</span><span class="p">,</span> <span class="n">E_surv_train</span><span class="p">)</span>
 </pre></div>

diff --git a/examples/user-guide/content/1-useage.ipynb b/examples/user-guide/content/1-useage.ipynb
@@ -249,8 +249,8 @@
     "X_surv_train, X_surv_test, Y_surv_train, Y_surv_test = train_test_split(X, Y, test_size=0.2)\n",
     "\n",
     "# introduce administrative censoring to simulate survival data\n",
-    "T_surv_train = np.minimum(Y_train, 30) # time of an event or censoring\n",
-    "E_surv_train = Y_train > 30 # 1 if T[i] is the time of an event, 0 if it's a time of censoring\n",
+    "T_surv_train = np.minimum(Y_surv_train, 30) # time of an event or censoring\n",
+    "E_surv_train = Y_surv_train > 30 # 1 if T[i] is the time of an event, 0 if it's a time of censoring\n",
     "\n",
     "ngb = NGBSurvival(Dist=LogNormal).fit(X_surv_train, T_surv_train, E_surv_train)"
    ]
@@ -632,7 +632,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.6"
+   "version": "3.7.7"
   }
  },
  "nbformat": 4,