{ "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "# Классификация" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### Загрузка набора данных" ] }, { "cell_type": "code", "execution_count": 1, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", " | Survived | \n", "Pclass | \n", "Name | \n", "Sex | \n", "Age | \n", "SibSp | \n", "Parch | \n", "Ticket | \n", "Fare | \n", "Cabin | \n", "Embarked | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|
PassengerId | \n", "\n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " |
1 | \n", "0 | \n", "3 | \n", "Braund, Mr. Owen Harris | \n", "male | \n", "22.0 | \n", "1 | \n", "0 | \n", "A/5 21171 | \n", "7.2500 | \n", "NaN | \n", "S | \n", "
2 | \n", "1 | \n", "1 | \n", "Cumings, Mrs. John Bradley (Florence Briggs Th... | \n", "female | \n", "38.0 | \n", "1 | \n", "0 | \n", "PC 17599 | \n", "71.2833 | \n", "C85 | \n", "C | \n", "
3 | \n", "1 | \n", "3 | \n", "Heikkinen, Miss. Laina | \n", "female | \n", "26.0 | \n", "0 | \n", "0 | \n", "STON/O2. 3101282 | \n", "7.9250 | \n", "NaN | \n", "S | \n", "
4 | \n", "1 | \n", "1 | \n", "Futrelle, Mrs. Jacques Heath (Lily May Peel) | \n", "female | \n", "35.0 | \n", "1 | \n", "0 | \n", "113803 | \n", "53.1000 | \n", "C123 | \n", "S | \n", "
5 | \n", "0 | \n", "3 | \n", "Allen, Mr. William Henry | \n", "male | \n", "35.0 | \n", "0 | \n", "0 | \n", "373450 | \n", "8.0500 | \n", "NaN | \n", "S | \n", "
... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "
887 | \n", "0 | \n", "2 | \n", "Montvila, Rev. Juozas | \n", "male | \n", "27.0 | \n", "0 | \n", "0 | \n", "211536 | \n", "13.0000 | \n", "NaN | \n", "S | \n", "
888 | \n", "1 | \n", "1 | \n", "Graham, Miss. Margaret Edith | \n", "female | \n", "19.0 | \n", "0 | \n", "0 | \n", "112053 | \n", "30.0000 | \n", "B42 | \n", "S | \n", "
889 | \n", "0 | \n", "3 | \n", "Johnston, Miss. Catherine Helen \"Carrie\" | \n", "female | \n", "NaN | \n", "1 | \n", "2 | \n", "W./C. 6607 | \n", "23.4500 | \n", "NaN | \n", "S | \n", "
890 | \n", "1 | \n", "1 | \n", "Behr, Mr. Karl Howell | \n", "male | \n", "26.0 | \n", "0 | \n", "0 | \n", "111369 | \n", "30.0000 | \n", "C148 | \n", "C | \n", "
891 | \n", "0 | \n", "3 | \n", "Dooley, Mr. Patrick | \n", "male | \n", "32.0 | \n", "0 | \n", "0 | \n", "370376 | \n", "7.7500 | \n", "NaN | \n", "Q | \n", "
891 rows × 11 columns
\n", "\n", " | Survived | \n", "Pclass | \n", "Name | \n", "Sex | \n", "Age | \n", "SibSp | \n", "Parch | \n", "Ticket | \n", "Fare | \n", "Cabin | \n", "Embarked | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|
PassengerId | \n", "\n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " |
145 | \n", "0 | \n", "2 | \n", "Andrew, Mr. Edgardo Samuel | \n", "male | \n", "18.00 | \n", "0 | \n", "0 | \n", "231945 | \n", "11.5000 | \n", "NaN | \n", "S | \n", "
206 | \n", "0 | \n", "3 | \n", "Strom, Miss. Telma Matilda | \n", "female | \n", "2.00 | \n", "0 | \n", "1 | \n", "347054 | \n", "10.4625 | \n", "G6 | \n", "S | \n", "
349 | \n", "1 | \n", "3 | \n", "Coutts, Master. William Loch \"William\" | \n", "male | \n", "3.00 | \n", "1 | \n", "1 | \n", "C.A. 37671 | \n", "15.9000 | \n", "NaN | \n", "S | \n", "
329 | \n", "1 | \n", "3 | \n", "Goldsmith, Mrs. Frank John (Emily Alice Brown) | \n", "female | \n", "31.00 | \n", "1 | \n", "1 | \n", "363291 | \n", "20.5250 | \n", "NaN | \n", "S | \n", "
289 | \n", "1 | \n", "2 | \n", "Hosono, Mr. Masabumi | \n", "male | \n", "42.00 | \n", "0 | \n", "0 | \n", "237798 | \n", "13.0000 | \n", "NaN | \n", "S | \n", "
... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "
756 | \n", "1 | \n", "2 | \n", "Hamalainen, Master. Viljo | \n", "male | \n", "0.67 | \n", "1 | \n", "1 | \n", "250649 | \n", "14.5000 | \n", "NaN | \n", "S | \n", "
816 | \n", "0 | \n", "1 | \n", "Fry, Mr. Richard | \n", "male | \n", "NaN | \n", "0 | \n", "0 | \n", "112058 | \n", "0.0000 | \n", "B102 | \n", "S | \n", "
890 | \n", "1 | \n", "1 | \n", "Behr, Mr. Karl Howell | \n", "male | \n", "26.00 | \n", "0 | \n", "0 | \n", "111369 | \n", "30.0000 | \n", "C148 | \n", "C | \n", "
738 | \n", "1 | \n", "1 | \n", "Lesurer, Mr. Gustave J | \n", "male | \n", "35.00 | \n", "0 | \n", "0 | \n", "PC 17755 | \n", "512.3292 | \n", "B101 | \n", "C | \n", "
61 | \n", "0 | \n", "3 | \n", "Sirayanian, Mr. Orsen | \n", "male | \n", "22.00 | \n", "0 | \n", "0 | \n", "2669 | \n", "7.2292 | \n", "NaN | \n", "C | \n", "
712 rows × 11 columns
\n", "\n", " | Survived | \n", "
---|---|
PassengerId | \n", "\n", " |
145 | \n", "0 | \n", "
206 | \n", "0 | \n", "
349 | \n", "1 | \n", "
329 | \n", "1 | \n", "
289 | \n", "1 | \n", "
... | \n", "... | \n", "
756 | \n", "1 | \n", "
816 | \n", "0 | \n", "
890 | \n", "1 | \n", "
738 | \n", "1 | \n", "
61 | \n", "0 | \n", "
712 rows × 1 columns
\n", "\n", " | Survived | \n", "Pclass | \n", "Name | \n", "Sex | \n", "Age | \n", "SibSp | \n", "Parch | \n", "Ticket | \n", "Fare | \n", "Cabin | \n", "Embarked | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|
PassengerId | \n", "\n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " |
843 | \n", "1 | \n", "1 | \n", "Serepeca, Miss. Augusta | \n", "female | \n", "30.0 | \n", "0 | \n", "0 | \n", "113798 | \n", "31.0000 | \n", "NaN | \n", "C | \n", "
791 | \n", "0 | \n", "3 | \n", "Keane, Mr. Andrew \"Andy\" | \n", "male | \n", "NaN | \n", "0 | \n", "0 | \n", "12460 | \n", "7.7500 | \n", "NaN | \n", "Q | \n", "
509 | \n", "0 | \n", "3 | \n", "Olsen, Mr. Henry Margido | \n", "male | \n", "28.0 | \n", "0 | \n", "0 | \n", "C 4001 | \n", "22.5250 | \n", "NaN | \n", "S | \n", "
828 | \n", "1 | \n", "2 | \n", "Mallet, Master. Andre | \n", "male | \n", "1.0 | \n", "0 | \n", "2 | \n", "S.C./PARIS 2079 | \n", "37.0042 | \n", "NaN | \n", "C | \n", "
414 | \n", "0 | \n", "2 | \n", "Cunningham, Mr. Alfred Fleming | \n", "male | \n", "NaN | \n", "0 | \n", "0 | \n", "239853 | \n", "0.0000 | \n", "NaN | \n", "S | \n", "
... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "
824 | \n", "1 | \n", "3 | \n", "Moor, Mrs. (Beila) | \n", "female | \n", "27.0 | \n", "0 | \n", "1 | \n", "392096 | \n", "12.4750 | \n", "E121 | \n", "S | \n", "
353 | \n", "0 | \n", "3 | \n", "Elias, Mr. Tannous | \n", "male | \n", "15.0 | \n", "1 | \n", "1 | \n", "2695 | \n", "7.2292 | \n", "NaN | \n", "C | \n", "
674 | \n", "1 | \n", "2 | \n", "Wilhelms, Mr. Charles | \n", "male | \n", "31.0 | \n", "0 | \n", "0 | \n", "244270 | \n", "13.0000 | \n", "NaN | \n", "S | \n", "
100 | \n", "0 | \n", "2 | \n", "Kantor, Mr. Sinai | \n", "male | \n", "34.0 | \n", "1 | \n", "0 | \n", "244367 | \n", "26.0000 | \n", "NaN | \n", "S | \n", "
542 | \n", "0 | \n", "3 | \n", "Andersson, Miss. Ingeborg Constanzia | \n", "female | \n", "9.0 | \n", "4 | \n", "2 | \n", "347082 | \n", "31.2750 | \n", "NaN | \n", "S | \n", "
179 rows × 11 columns
\n", "\n", " | Survived | \n", "
---|---|
PassengerId | \n", "\n", " |
843 | \n", "1 | \n", "
791 | \n", "0 | \n", "
509 | \n", "0 | \n", "
828 | \n", "1 | \n", "
414 | \n", "0 | \n", "
... | \n", "... | \n", "
824 | \n", "1 | \n", "
353 | \n", "0 | \n", "
674 | \n", "1 | \n", "
100 | \n", "0 | \n", "
542 | \n", "0 | \n", "
179 rows × 1 columns
\n", "\n", " | Cabin_type_B | \n", "Cabin_type_C | \n", "Cabin_type_D | \n", "Cabin_type_E | \n", "Cabin_type_F | \n", "Cabin_type_G | \n", "Cabin_type_T | \n", "Cabin_type_u | \n", "Is_married | \n", "Pclass | \n", "Age | \n", "SibSp | \n", "Sex_male | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|---|---|
PassengerId | \n", "\n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " |
145 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "1.0 | \n", "0 | \n", "-0.379423 | \n", "-0.869506 | \n", "-0.473465 | \n", "1.0 | \n", "
206 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "1.0 | \n", "0.0 | \n", "0.0 | \n", "0 | \n", "0.821241 | \n", "-2.102186 | \n", "-0.473465 | \n", "0.0 | \n", "
349 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "1.0 | \n", "0 | \n", "0.821241 | \n", "-2.025143 | \n", "0.437635 | \n", "1.0 | \n", "
329 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "1.0 | \n", "1 | \n", "0.821241 | \n", "0.132047 | \n", "0.437635 | \n", "0.0 | \n", "
289 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "1.0 | \n", "0 | \n", "-0.379423 | \n", "0.979514 | \n", "-0.473465 | \n", "1.0 | \n", "
... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "
756 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "1.0 | \n", "0 | \n", "-0.379423 | \n", "-2.204652 | \n", "0.437635 | \n", "1.0 | \n", "
816 | \n", "1.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0 | \n", "-1.580088 | \n", "-0.099081 | \n", "-0.473465 | \n", "1.0 | \n", "
890 | \n", "0.0 | \n", "1.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0 | \n", "-1.580088 | \n", "-0.253166 | \n", "-0.473465 | \n", "1.0 | \n", "
738 | \n", "1.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0 | \n", "-1.580088 | \n", "0.440217 | \n", "-0.473465 | \n", "1.0 | \n", "
61 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "1.0 | \n", "0 | \n", "0.821241 | \n", "-0.561336 | \n", "-0.473465 | \n", "1.0 | \n", "
712 rows × 13 columns
\n", "\n", " | Precision_train | \n", "Precision_test | \n", "Recall_train | \n", "Recall_test | \n", "Accuracy_train | \n", "Accuracy_test | \n", "F1_train | \n", "F1_test | \n", "
---|---|---|---|---|---|---|---|---|
random_forest | \n", "0.894340 | \n", "0.794118 | \n", "0.868132 | \n", "0.782609 | \n", "0.910112 | \n", "0.837989 | \n", "0.881041 | \n", "0.788321 | \n", "
gradient_boosting | \n", "0.889764 | \n", "0.800000 | \n", "0.827839 | \n", "0.753623 | \n", "0.894663 | \n", "0.832402 | \n", "0.857685 | \n", "0.776119 | \n", "
logistic | \n", "0.751880 | \n", "0.806452 | \n", "0.732601 | \n", "0.724638 | \n", "0.804775 | \n", "0.826816 | \n", "0.742115 | \n", "0.763359 | \n", "
decision_tree | \n", "0.852459 | \n", "0.839286 | \n", "0.761905 | \n", "0.681159 | \n", "0.858146 | \n", "0.826816 | \n", "0.804642 | \n", "0.752000 | \n", "
knn | \n", "0.829167 | \n", "0.827586 | \n", "0.728938 | \n", "0.695652 | \n", "0.838483 | \n", "0.826816 | \n", "0.775828 | \n", "0.755906 | \n", "
ridge | \n", "0.720395 | \n", "0.688312 | \n", "0.802198 | \n", "0.768116 | \n", "0.804775 | \n", "0.776536 | \n", "0.759099 | \n", "0.726027 | \n", "
naive_bayes | \n", "0.554524 | \n", "0.575472 | \n", "0.875458 | \n", "0.884058 | \n", "0.682584 | \n", "0.703911 | \n", "0.678977 | \n", "0.697143 | \n", "
mlp | \n", "0.900000 | \n", "0.833333 | \n", "0.197802 | \n", "0.217391 | \n", "0.683989 | \n", "0.681564 | \n", "0.324324 | \n", "0.344828 | \n", "
\n", " | Accuracy_test | \n", "F1_test | \n", "ROC_AUC_test | \n", "Cohen_kappa_test | \n", "MCC_test | \n", "
---|---|---|---|---|---|
random_forest | \n", "0.837989 | \n", "0.788321 | \n", "0.858893 | \n", "0.657111 | \n", "0.657157 | \n", "
logistic | \n", "0.826816 | \n", "0.763359 | \n", "0.854084 | \n", "0.627409 | \n", "0.629641 | \n", "
ridge | \n", "0.776536 | \n", "0.726027 | \n", "0.851054 | \n", "0.538303 | \n", "0.540613 | \n", "
gradient_boosting | \n", "0.832402 | \n", "0.776119 | \n", "0.850922 | \n", "0.642381 | \n", "0.643113 | \n", "
knn | \n", "0.826816 | \n", "0.755906 | \n", "0.838735 | \n", "0.623260 | \n", "0.628905 | \n", "
decision_tree | \n", "0.826816 | \n", "0.752000 | \n", "0.794137 | \n", "0.621151 | \n", "0.629142 | \n", "
naive_bayes | \n", "0.703911 | \n", "0.697143 | \n", "0.785903 | \n", "0.431814 | \n", "0.470403 | \n", "
mlp | \n", "0.681564 | \n", "0.344828 | \n", "0.712714 | \n", "0.220490 | \n", "0.307678 | \n", "
\n", " | Survived | \n", "Predicted | \n", "Pclass | \n", "Name | \n", "Sex | \n", "Age | \n", "SibSp | \n", "Parch | \n", "Ticket | \n", "Fare | \n", "Cabin | \n", "Embarked | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|---|
PassengerId | \n", "\n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " |
26 | \n", "1 | \n", "0 | \n", "3 | \n", "Asplund, Mrs. Carl Oscar (Selma Augusta Emilia... | \n", "female | \n", "38.0 | \n", "1 | \n", "5 | \n", "347077 | \n", "31.3875 | \n", "NaN | \n", "S | \n", "
72 | \n", "0 | \n", "1 | \n", "3 | \n", "Goodwin, Miss. Lillian Amy | \n", "female | \n", "16.0 | \n", "5 | \n", "2 | \n", "CA 2144 | \n", "46.9000 | \n", "NaN | \n", "S | \n", "
103 | \n", "0 | \n", "1 | \n", "1 | \n", "White, Mr. Richard Frasar | \n", "male | \n", "21.0 | \n", "0 | \n", "1 | \n", "35281 | \n", "77.2875 | \n", "D26 | \n", "S | \n", "
108 | \n", "1 | \n", "0 | \n", "3 | \n", "Moss, Mr. Albert Johan | \n", "male | \n", "NaN | \n", "0 | \n", "0 | \n", "312991 | \n", "7.7750 | \n", "NaN | \n", "S | \n", "
128 | \n", "1 | \n", "0 | \n", "3 | \n", "Madsen, Mr. Fridtjof Arne | \n", "male | \n", "24.0 | \n", "0 | \n", "0 | \n", "C 17369 | \n", "7.1417 | \n", "NaN | \n", "S | \n", "
193 | \n", "1 | \n", "0 | \n", "3 | \n", "Andersen-Jensen, Miss. Carla Christine Nielsine | \n", "female | \n", "19.0 | \n", "1 | \n", "0 | \n", "350046 | \n", "7.8542 | \n", "NaN | \n", "S | \n", "
241 | \n", "0 | \n", "1 | \n", "3 | \n", "Zabour, Miss. Thamine | \n", "female | \n", "NaN | \n", "1 | \n", "0 | \n", "2665 | \n", "14.4542 | \n", "NaN | \n", "C | \n", "
272 | \n", "1 | \n", "0 | \n", "3 | \n", "Tornquist, Mr. William Henry | \n", "male | \n", "25.0 | \n", "0 | \n", "0 | \n", "LINE | \n", "0.0000 | \n", "NaN | \n", "S | \n", "
293 | \n", "0 | \n", "1 | \n", "2 | \n", "Levy, Mr. Rene Jacques | \n", "male | \n", "36.0 | \n", "0 | \n", "0 | \n", "SC/Paris 2163 | \n", "12.8750 | \n", "D | \n", "C | \n", "
352 | \n", "0 | \n", "1 | \n", "1 | \n", "Williams-Lambert, Mr. Fletcher Fellows | \n", "male | \n", "NaN | \n", "0 | \n", "0 | \n", "113510 | \n", "35.0000 | \n", "C128 | \n", "S | \n", "
358 | \n", "0 | \n", "1 | \n", "2 | \n", "Funk, Miss. Annie Clemmer | \n", "female | \n", "38.0 | \n", "0 | \n", "0 | \n", "237671 | \n", "13.0000 | \n", "NaN | \n", "S | \n", "
378 | \n", "0 | \n", "1 | \n", "1 | \n", "Widener, Mr. Harry Elkins | \n", "male | \n", "27.0 | \n", "0 | \n", "2 | \n", "113503 | \n", "211.5000 | \n", "C82 | \n", "C | \n", "
445 | \n", "1 | \n", "0 | \n", "3 | \n", "Johannesen-Bratthammer, Mr. Bernt | \n", "male | \n", "NaN | \n", "0 | \n", "0 | \n", "65306 | \n", "8.1125 | \n", "NaN | \n", "S | \n", "
450 | \n", "1 | \n", "0 | \n", "1 | \n", "Peuchen, Major. Arthur Godfrey | \n", "male | \n", "52.0 | \n", "0 | \n", "0 | \n", "113786 | \n", "30.5000 | \n", "C104 | \n", "S | \n", "
508 | \n", "1 | \n", "0 | \n", "1 | \n", "Bradley, Mr. George (\"George Arthur Brayton\") | \n", "male | \n", "NaN | \n", "0 | \n", "0 | \n", "111427 | \n", "26.5500 | \n", "NaN | \n", "S | \n", "
511 | \n", "1 | \n", "0 | \n", "3 | \n", "Daly, Mr. Eugene Patrick | \n", "male | \n", "29.0 | \n", "0 | \n", "0 | \n", "382651 | \n", "7.7500 | \n", "NaN | \n", "Q | \n", "
570 | \n", "1 | \n", "0 | \n", "3 | \n", "Jonsson, Mr. Carl | \n", "male | \n", "32.0 | \n", "0 | \n", "0 | \n", "350417 | \n", "7.8542 | \n", "NaN | \n", "S | \n", "
579 | \n", "0 | \n", "1 | \n", "3 | \n", "Caram, Mrs. Joseph (Maria Elias) | \n", "female | \n", "NaN | \n", "1 | \n", "0 | \n", "2689 | \n", "14.4583 | \n", "NaN | \n", "C | \n", "
584 | \n", "0 | \n", "1 | \n", "1 | \n", "Ross, Mr. John Hugo | \n", "male | \n", "36.0 | \n", "0 | \n", "0 | \n", "13049 | \n", "40.1250 | \n", "A10 | \n", "C | \n", "
588 | \n", "1 | \n", "0 | \n", "1 | \n", "Frolicher-Stehli, Mr. Maxmillian | \n", "male | \n", "60.0 | \n", "1 | \n", "1 | \n", "13567 | \n", "79.2000 | \n", "B41 | \n", "C | \n", "
618 | \n", "0 | \n", "1 | \n", "3 | \n", "Lobb, Mrs. William Arthur (Cordelia K Stanlick) | \n", "female | \n", "26.0 | \n", "1 | \n", "0 | \n", "A/5. 3336 | \n", "16.1000 | \n", "NaN | \n", "S | \n", "
658 | \n", "0 | \n", "1 | \n", "3 | \n", "Bourke, Mrs. John (Catherine) | \n", "female | \n", "32.0 | \n", "1 | \n", "1 | \n", "364849 | \n", "15.5000 | \n", "NaN | \n", "Q | \n", "
661 | \n", "1 | \n", "0 | \n", "1 | \n", "Frauenthal, Dr. Henry William | \n", "male | \n", "50.0 | \n", "2 | \n", "0 | \n", "PC 17611 | \n", "133.6500 | \n", "NaN | \n", "S | \n", "
674 | \n", "1 | \n", "0 | \n", "2 | \n", "Wilhelms, Mr. Charles | \n", "male | \n", "31.0 | \n", "0 | \n", "0 | \n", "244270 | \n", "13.0000 | \n", "NaN | \n", "S | \n", "
745 | \n", "1 | \n", "0 | \n", "3 | \n", "Stranden, Mr. Juho | \n", "male | \n", "31.0 | \n", "0 | \n", "0 | \n", "STON/O 2. 3101288 | \n", "7.9250 | \n", "NaN | \n", "S | \n", "
773 | \n", "0 | \n", "1 | \n", "2 | \n", "Mack, Mrs. (Mary) | \n", "female | \n", "57.0 | \n", "0 | \n", "0 | \n", "S.O./P.P. 3 | \n", "10.5000 | \n", "E77 | \n", "S | \n", "
807 | \n", "0 | \n", "1 | \n", "1 | \n", "Andrews, Mr. Thomas Jr | \n", "male | \n", "39.0 | \n", "0 | \n", "0 | \n", "112050 | \n", "0.0000 | \n", "A36 | \n", "S | \n", "
814 | \n", "0 | \n", "1 | \n", "3 | \n", "Andersson, Miss. Ebba Iris Alfrida | \n", "female | \n", "6.0 | \n", "4 | \n", "2 | \n", "347082 | \n", "31.2750 | \n", "NaN | \n", "S | \n", "
829 | \n", "1 | \n", "0 | \n", "3 | \n", "McCormack, Mr. Thomas Joseph | \n", "male | \n", "NaN | \n", "0 | \n", "0 | \n", "367228 | \n", "7.7500 | \n", "NaN | \n", "Q | \n", "
\n", " | Survived | \n", "Pclass | \n", "Name | \n", "Sex | \n", "Age | \n", "SibSp | \n", "Parch | \n", "Ticket | \n", "Fare | \n", "Cabin | \n", "Embarked | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|
450 | \n", "1 | \n", "1 | \n", "Peuchen, Major. Arthur Godfrey | \n", "male | \n", "52.0 | \n", "0 | \n", "0 | \n", "113786 | \n", "30.5 | \n", "C104 | \n", "S | \n", "
\n", " | Cabin_type_B | \n", "Cabin_type_C | \n", "Cabin_type_D | \n", "Cabin_type_E | \n", "Cabin_type_F | \n", "Cabin_type_G | \n", "Cabin_type_T | \n", "Cabin_type_u | \n", "Is_married | \n", "Pclass | \n", "Age | \n", "SibSp | \n", "Sex_male | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|---|---|
450 | \n", "0.0 | \n", "1.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "0.0 | \n", "-1.580088 | \n", "1.749939 | \n", "-0.473465 | \n", "1.0 | \n", "
\n", " | Precision_train | \n", "Precision_test | \n", "Recall_train | \n", "Recall_test | \n", "Accuracy_train | \n", "Accuracy_test | \n", "F1_train | \n", "F1_test | \n", "
---|---|---|---|---|---|---|---|---|
Name | \n", "\n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " | \n", " |
Old | \n", "0.894340 | \n", "0.794118 | \n", "0.868132 | \n", "0.782609 | \n", "0.910112 | \n", "0.837989 | \n", "0.881041 | \n", "0.788321 | \n", "
New | \n", "0.800699 | \n", "0.777778 | \n", "0.838828 | \n", "0.811594 | \n", "0.858146 | \n", "0.837989 | \n", "0.819320 | \n", "0.794326 | \n", "
\n", " | Accuracy_test | \n", "F1_test | \n", "ROC_AUC_test | \n", "Cohen_kappa_test | \n", "MCC_test | \n", "
---|---|---|---|---|---|
Name | \n", "\n", " | \n", " | \n", " | \n", " | \n", " |
Old | \n", "0.837989 | \n", "0.788321 | \n", "0.858893 | \n", "0.657111 | \n", "0.657157 | \n", "
New | \n", "0.837989 | \n", "0.794326 | \n", "0.866140 | \n", "0.660785 | \n", "0.661193 | \n", "
\n", " | T | \n", "Al2O3 | \n", "TiO2 | \n", "Density | \n", "
---|---|---|---|---|
0 | \n", "20 | \n", "0.0 | \n", "0.0 | \n", "1.06250 | \n", "
1 | \n", "25 | \n", "0.0 | \n", "0.0 | \n", "1.05979 | \n", "
2 | \n", "35 | \n", "0.0 | \n", "0.0 | \n", "1.05404 | \n", "
\n", " | T | \n", "Al2O3 | \n", "TiO2 | \n", "Density | \n", "
---|---|---|---|---|
0 | \n", "30 | \n", "0.00 | \n", "0.0 | \n", "1.05696 | \n", "
1 | \n", "55 | \n", "0.00 | \n", "0.0 | \n", "1.04158 | \n", "
2 | \n", "25 | \n", "0.05 | \n", "0.0 | \n", "1.08438 | \n", "
\n", " | T | \n", "Al2O3 | \n", "TiO2 | \n", "
---|---|---|---|
0 | \n", "20 | \n", "0.0 | \n", "0.0 | \n", "
1 | \n", "25 | \n", "0.0 | \n", "0.0 | \n", "
2 | \n", "35 | \n", "0.0 | \n", "0.0 | \n", "
\n", " | Density | \n", "
---|---|
0 | \n", "1.06250 | \n", "
1 | \n", "1.05979 | \n", "
2 | \n", "1.05404 | \n", "
\n", " | T | \n", "Al2O3 | \n", "TiO2 | \n", "
---|---|---|---|
0 | \n", "30 | \n", "0.00 | \n", "0.0 | \n", "
1 | \n", "55 | \n", "0.00 | \n", "0.0 | \n", "
2 | \n", "25 | \n", "0.05 | \n", "0.0 | \n", "
\n", " | Density | \n", "
---|---|
0 | \n", "1.05696 | \n", "
1 | \n", "1.04158 | \n", "
2 | \n", "1.08438 | \n", "
\n", " | RMSE_train | \n", "RMSE_test | \n", "RMAE_test | \n", "R2_test | \n", "
---|---|---|---|---|
linear_poly | \n", "0.000319 | \n", "0.000362 | \n", "0.016643 | \n", "0.999965 | \n", "
linear_interact | \n", "0.001131 | \n", "0.001491 | \n", "0.033198 | \n", "0.999413 | \n", "
linear | \n", "0.002464 | \n", "0.003261 | \n", "0.049891 | \n", "0.997191 | \n", "
random_forest | \n", "0.002716 | \n", "0.005575 | \n", "0.067298 | \n", "0.991788 | \n", "
decision_tree | \n", "0.000346 | \n", "0.006433 | \n", "0.076138 | \n", "0.989067 | \n", "
ridge | \n", "0.013989 | \n", "0.015356 | \n", "0.116380 | \n", "0.937703 | \n", "
knn | \n", "0.053108 | \n", "0.056776 | \n", "0.217611 | \n", "0.148414 | \n", "
mlp | \n", "0.095734 | \n", "0.099654 | \n", "0.270371 | \n", "-1.623554 | \n", "
\n", " | T | \n", "Al2O3 | \n", "TiO2 | \n", "Density | \n", "DensityPred | \n", "
---|---|---|---|---|---|
0 | \n", "20 | \n", "0.0 | \n", "0.0 | \n", "1.06250 | \n", "1.063174 | \n", "
1 | \n", "25 | \n", "0.0 | \n", "0.0 | \n", "1.05979 | \n", "1.060117 | \n", "
2 | \n", "35 | \n", "0.0 | \n", "0.0 | \n", "1.05404 | \n", "1.053941 | \n", "
3 | \n", "40 | \n", "0.0 | \n", "0.0 | \n", "1.05103 | \n", "1.050822 | \n", "
4 | \n", "45 | \n", "0.0 | \n", "0.0 | \n", "1.04794 | \n", "1.047683 | \n", "
\n", " | T | \n", "Al2O3 | \n", "TiO2 | \n", "Density | \n", "DensityPred | \n", "
---|---|---|---|---|---|
0 | \n", "30 | \n", "0.00 | \n", "0.0 | \n", "1.05696 | \n", "1.057040 | \n", "
1 | \n", "55 | \n", "0.00 | \n", "0.0 | \n", "1.04158 | \n", "1.041341 | \n", "
2 | \n", "25 | \n", "0.05 | \n", "0.0 | \n", "1.08438 | \n", "1.084063 | \n", "
3 | \n", "30 | \n", "0.05 | \n", "0.0 | \n", "1.08112 | \n", "1.080764 | \n", "
4 | \n", "35 | \n", "0.05 | \n", "0.0 | \n", "1.07781 | \n", "1.077444 | \n", "