Zero_Shot_Classification

Author: anasjawed283

zeroshot_classification_NLP.ipynb

@@ -0,0 +1,115 @@
+{
+  "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": 2,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "h-TO-Q8HnNSI",
+        "outputId": "cf9f170f-b8af-4f3d-877f-da9641ca7ff7"
+      },
+      "outputs": [
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            "Requirement already satisfied: transformers in /usr/local/lib/python3.10/dist-packages (4.44.2)\n",
+            "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from transformers) (3.16.1)\n",
+            "Requirement already satisfied: huggingface-hub<1.0,>=0.23.2 in /usr/local/lib/python3.10/dist-packages (from transformers) (0.24.7)\n",
+            "Requirement already satisfied: numpy>=1.17 in /usr/local/lib/python3.10/dist-packages (from transformers) (1.26.4)\n",
+            "Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.10/dist-packages (from transformers) (24.1)\n",
+            "Requirement already satisfied: pyyaml>=5.1 in /usr/local/lib/python3.10/dist-packages (from transformers) (6.0.2)\n",
+            "Requirement already satisfied: regex!=2019.12.17 in /usr/local/lib/python3.10/dist-packages (from transformers) (2024.9.11)\n",
+            "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from transformers) (2.32.3)\n",
+            "Requirement already satisfied: safetensors>=0.4.1 in /usr/local/lib/python3.10/dist-packages (from transformers) (0.4.5)\n",
+            "Requirement already satisfied: tokenizers<0.20,>=0.19 in /usr/local/lib/python3.10/dist-packages (from transformers) (0.19.1)\n",
+            "Requirement already satisfied: tqdm>=4.27 in /usr/local/lib/python3.10/dist-packages (from transformers) (4.66.5)\n",
+            "Requirement already satisfied: fsspec>=2023.5.0 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.23.2->transformers) (2024.6.1)\n",
+            "Requirement already satisfied: typing-extensions>=3.7.4.3 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.23.2->transformers) (4.12.2)\n",
+            "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->transformers) (3.4.0)\n",
+            "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->transformers) (3.10)\n",
+            "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->transformers) (2.2.3)\n",
+            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->transformers) (2024.8.30)\n"
+          ]
+        }
+      ],
+      "source": [
+        "# Import necessary libraries\n",
+        "!pip install transformers\n",
+        "\n",
+        "from transformers import pipeline\n",
+        "\n",
+        "# Initialize the zero-shot classifier pipeline\n",
+        "classifier = pipeline(\"zero-shot-classification\", model=\"facebook/bart-large-mnli\")\n",
+        "\n",
+        "# Define the labels/categories for classification\n",
+        "candidate_labels = [\n",
+        "    \"sports\", \"education\", \"entertainment\", \"harassment\", \"violence\",\n",
+        "    \"kids\", \"politics\", \"technology\", \"history\", \"geography\", \"economics\",\n",
+        "    \"health\", \"science\", \"environment\", \"culture\"\n",
+        "]"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 3,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "pt9YEVe5nu9_",
+        "outputId": "6e5f63d3-652d-46bb-d4fb-8f6033afec33"
+      },
+      "outputs": [
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            "politics: 0.59\n",
+            "history: 0.15\n",
+            "culture: 0.05\n",
+            "technology: 0.03\n",
+            "entertainment: 0.03\n",
+            "environment: 0.02\n",
+            "violence: 0.02\n",
+            "kids: 0.02\n",
+            "geography: 0.02\n",
+            "health: 0.02\n",
+            "economics: 0.02\n",
+            "science: 0.01\n",
+            "harassment: 0.01\n",
+            "sports: 0.01\n",
+            "education: 0.01\n"
+          ]
+        }
+      ],
+      "source": [
+        "# Example text for classification\n",
+        "text = \"The member won the presidency election by 75 seats.\"\n",
+        "\n",
+        "# Perform zero-shot classification\n",
+        "result = classifier(text, candidate_labels)\n",
+        "\n",
+        "# Display the classification result\n",
+        "for label, score in zip(result['labels'], result['scores']):\n",
+        "    print(f\"{label}: {score:.2f}\")\n"
+      ]
+    }
+  ],
+  "metadata": {
+    "colab": {
+      "provenance": []
+    },
+    "kernelspec": {
+      "display_name": "Python 3",
+      "name": "python3"
+    },
+    "language_info": {
+      "name": "python"
+    }
+  },
+  "nbformat": 4,
+  "nbformat_minor": 0
+}