Update Titanic Random Forest Tensorflow.ipynb

wncc · Oct 17, 2020 · 41662b4 · 41662b4
1 parent 3365f40
commit 41662b4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Titanic Random Forest Tensorflow.ipynb b/Titanic Random Forest Tensorflow.ipynb
@@ -1671,6 +1671,8 @@
    "outputs": [],
    "source": [
     "from sklearn.ensemble import RandomForestClassifier\n",
+    "clf=RandomForestClassifier(n_estimator=8).fit(train, y_train)\n",
+    "predictions = clf.predict(test)"
     "#Write code for Random Forests here"
    ]
   },