diff --git a/Model/drug_pipeline.skops b/Model/drug_pipeline.skops index eeb0a39..5d13786 100644 Binary files a/Model/drug_pipeline.skops and b/Model/drug_pipeline.skops differ diff --git a/Results/metrics.txt b/Results/metrics.txt index 15db8b9..0f8fb31 100644 --- a/Results/metrics.txt +++ b/Results/metrics.txt @@ -1,2 +1,2 @@ -Accuracy = 0.97, F1 Score = 0.94. \ No newline at end of file +Accuracy = 0.98, F1 Score = 0.95. \ No newline at end of file diff --git a/Results/model_results.png b/Results/model_results.png index 9f30bf7..f2236e8 100644 Binary files a/Results/model_results.png and b/Results/model_results.png differ diff --git a/notebook.ipynb b/notebook.ipynb index f0ee2b0..06181da 100644 --- a/notebook.ipynb +++ b/notebook.ipynb @@ -2,7 +2,7 @@ "cells": [ { "cell_type": "code", - "execution_count": 1, + "execution_count": 30, "id": "57bbae0c", "metadata": {}, "outputs": [ @@ -37,31 +37,31 @@ " \n", " \n", " \n", - " 100\n", - " 31\n", - " M\n", - " HIGH\n", + " 72\n", + " 24\n", + " F\n", " NORMAL\n", - " 11.871\n", - " drugA\n", + " HIGH\n", + " 10.605\n", + " drugX\n", " \n", " \n", - " 8\n", - " 60\n", - " M\n", + " 139\n", + " 69\n", + " F\n", " NORMAL\n", " HIGH\n", - " 15.171\n", - " DrugY\n", + " 10.065\n", + " drugX\n", " \n", " \n", - " 23\n", - " 48\n", - " F\n", - " LOW\n", + " 56\n", + " 65\n", + " M\n", " HIGH\n", - " 15.036\n", - " DrugY\n", + " NORMAL\n", + " 11.340\n", + " drugB\n", " \n", " \n", "\n", @@ -69,12 +69,12 @@ ], "text/plain": [ " Age Sex BP Cholesterol Na_to_K Drug\n", - "100 31 M HIGH NORMAL 11.871 drugA\n", - "8 60 M NORMAL HIGH 15.171 DrugY\n", - "23 48 F LOW HIGH 15.036 DrugY" + "72 24 F NORMAL HIGH 10.605 drugX\n", + "139 69 F NORMAL HIGH 10.065 drugX\n", + "56 65 M HIGH NORMAL 11.340 drugB" ] }, - "execution_count": 1, + "execution_count": 30, "metadata": {}, "output_type": "execute_result" } @@ -89,7 +89,7 @@ }, { "cell_type": "code", - "execution_count": 2, + "execution_count": 31, "id": "b32a6d5d", "metadata": {}, "outputs": [], @@ -106,14 +106,14 @@ }, { "cell_type": "code", - "execution_count": 3, + "execution_count": 32, "id": "0d008cb7", "metadata": {}, "outputs": [ { "data": { "text/html": [ - "
Pipeline(steps=[('preprocessing',\n",
+       "
Pipeline(steps=[('preprocessing',\n",
        "                 ColumnTransformer(transformers=[('encoder', OrdinalEncoder(),\n",
        "                                                  [1, 2, 3]),\n",
        "                                                 ('num_imputer',\n",
@@ -671,7 +671,7 @@
        "                                                  [0, 4]),\n",
        "                                                 ('num_scaler',\n",
        "                                                  StandardScaler(), [0, 4])])),\n",
-       "                ('model', RandomForestClassifier(random_state=125))])
In a Jupyter environment, please rerun this cell to show the HTML representation or trust the notebook. 
On GitHub, the HTML representation is unable to render, please try loading this page with nbviewer.org.
In a Jupyter environment, please rerun this cell to show the HTML representation or trust the notebook.
On GitHub, the HTML representation is unable to render, please try loading this page with nbviewer.org.