{ "cells": [ { "cell_type": "markdown", "metadata": { "colab_type": "text", "id": "HOAv4FLaAC3f" }, "source": [ "# Compute Recall Score for a Classification Model" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "colab": {}, "colab_type": "code", "id": "bN_083VRAC3h" }, "outputs": [], "source": [ "# import libraries\n", "import pandas as pd\n", "from sklearn.model_selection import train_test_split\n", "from sklearn.linear_model import LogisticRegression\n", "\n", "import warnings\n", "warnings.filterwarnings(\"ignore\")" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "colab": { "base_uri": "https://localhost:8080/", "height": 194 }, "colab_type": "code", "id": "eMkdycnKAC3k", "outputId": "f76c521b-3730-4226-c498-36092769b6d5" }, "outputs": [], "source": [ "# data doesn't have headers, so let's create headers\n", "_headers = ['buying', 'maint', 'doors', 'persons', 'lug_boot', 'safety', 'car']\n", "# read in cars dataset\n", "df = pd.read_csv('../Dataset/car.data', names=_headers, index_col=None)\n", "df.head()\n", "\n", "# target column is 'car'" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "colab": { "base_uri": "https://localhost:8080/", "height": 214 }, "colab_type": "code", "id": "EQvotJcIAC3o", "outputId": "0f660e88-220f-4d0b-8b4a-3872d61d2672" }, "outputs": [], "source": [ "# encode categorical variables\n", "_df = pd.get_dummies(df, columns=['buying', 'maint', 'doors', 'persons', 'lug_boot', 'safety'])\n", "_df.head()" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "colab": {}, "colab_type": "code", "id": "JcMh8CFzAC3q" }, "outputs": [], "source": [ "# target column is 'car'\n", "\n", "features = _df.drop(['car'], axis=1).values\n", "labels = _df[['car']].values\n", "\n", "# split 80% for training and 20% into an evaluation set\n", "X_train, X_eval, y_train, y_eval = train_test_split(features, labels, test_size=0.3, random_state=0)\n", "\n", "# further split the evaluation set into validation and test sets of 10% each\n", "X_val, X_test, y_val, y_test = train_test_split(X_eval, y_eval, test_size=0.5, random_state=0)" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "colab": { "base_uri": "https://localhost:8080/", "height": 161 }, "colab_type": "code", "id": "-4nPRPmXAC3s", "outputId": "03ff35f5-119c-4120-8040-a38a8d144914" }, "outputs": [], "source": [ "# train a Logistic Regression model\n", "model = LogisticRegression()\n", "model.fit(X_train, y_train)" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "colab": {}, "colab_type": "code", "id": "KziGroVtAC3u" }, "outputs": [], "source": [ "# make predictions for the validation dataset\n", "y_pred = model.predict(X_val)" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "colab": {}, "colab_type": "code", "id": "YfpxsWw0AC3w" }, "outputs": [], "source": [ "# import libraries\n", "from sklearn.metrics import recall_score" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "colab": { "base_uri": "https://localhost:8080/", "height": 35 }, "colab_type": "code", "id": "phVn4D7cAC3z", "outputId": "6a92658f-c09e-4dda-f2b9-4fdfe7d1ea4e" }, "outputs": [], "source": [ "recall_score = recall_score(y_val, y_pred, average='macro')\n", "print(recall_score)" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "colab": {}, "colab_type": "code", "id": "MTlthhG0AC31" }, "outputs": [], "source": [] } ], "metadata": { "colab": { "name": "Exercise6_08.ipynb", "provenance": [] }, "file_extension": ".py", "kernelspec": { "display_name": "Python 3", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.8.6" }, "mimetype": "text/x-python", "name": "python", "npconvert_exporter": "python", "pygments_lexer": "ipython3", "version": 3 }, "nbformat": 4, "nbformat_minor": 1 }