classivax 0.1.0__tar.gz

classivax-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Luiz Felipe Fernandes Neves
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

classivax-0.1.0/PKG-INFO

@@ -0,0 +1,121 @@
+Metadata-Version: 2.4
+Name: classivax
+Version: 0.1.0
+Summary: Classify vaccine-related news headlines into topic categories using a Recognizing Textual Entailment (RTE) approach.
+Author: Luiz Felipe Fernandes Neves
+License: MIT
+Project-URL: Homepage, https://github.com/lffernandes08/classivax
+Project-URL: Repository, https://github.com/lffernandes08/classivax
+Project-URL: Model, https://huggingface.co/LuizNeves/BERT-NLI-vaccine-v2
+Project-URL: Paper, https://doi.org/10.5117/CCR2025.1.1.NEVE
+Keywords: nlp,text-classification,natural-language-inference,textual-entailment,vaccine,science-communication,computational-social-science
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: transformers>=4.30
+Requires-Dist: torch>=2.0
+Requires-Dist: pandas>=1.5
+Dynamic: license-file
+
+# ClassiVax
+
+Classify vaccine-related news headlines into topic categories using a **Recognizing Textual Entailment (RTE)** approach, powered by the [`LuizNeves/BERT-NLI-vaccine-v2`](https://huggingface.co/LuizNeves/BERT-NLI-vaccine-v2) model.
+
+Built on the method described in:
+
+> Neves, L. F. F., Camargo, C. Q., & Massarani, L. (2025). Boosting Transformers: Recognizing Textual Entailment for Classification of Vaccine News Coverage. *Computational Communication Research*, 7(1), 1–32. https://doi.org/10.5117/CCR2025.1.1.NEVE
+
+## Installation
+
+```bash
+pip install classivax
+```
+
+## Usage
+
+### Default classes (10 vaccine-related topics from the paper)
+
+```python
+from classivax import ClassiVaxClassifier
+
+clf = ClassiVaxClassifier()
+
+result = clf.classify("Twitter will begin removing vaccine misinformation")
+print(result["class"])   # "Vaccine Hesitancy and Mis-/Disinformation"
+print(result["score"])   # e.g. 0.999
+```
+
+### Batch classification
+
+```python
+headlines = [
+    "Twitter will begin removing vaccine misinformation",
+    "Britain begins coronavirus vaccine rollout",
+    "COVID-19 vaccination key to economic recovery",
+]
+
+df = clf.classify_batch(headlines)
+print(df)
+```
+
+### Custom classes
+
+The underlying RTE approach also supports classifying text against any set of natural-language class descriptions you define — not just the original 10 classes.
+
+```python
+custom_classes = {
+    "Vaccine Safety Debate": "The headline discusses concerns, controversies, or debates about vaccine safety",
+    "Vaccine Logistics": "The headline discusses shipping, storage, cold chain, or distribution logistics for vaccines",
+}
+
+clf = ClassiVaxClassifier(classes=custom_classes)
+result = clf.classify("Cold storage shortages delay vaccine shipments to rural areas")
+print(result["class"])  # "Vaccine Logistics"
+```
+
+**Note:** the underlying model was fine-tuned specifically on English-language, vaccine-related headlines from 2020–2021. Using custom classes outside this domain is supported, but accuracy has not been formally evaluated beyond the original study's scope — treat results on very different domains with appropriate caution.
+
+## The 10 default classes
+
+| Class | Description |
+|---|---|
+| Global Access to Vaccine | Initiatives for equal access to COVID-19 vaccines (e.g., COVAX), and/or the need to combat inequity, ensure global distribution, and contain the disease, especially in low- and middle-income countries |
+| Science and Technology | The science behind vaccine development, including research, studies, safety and efficacy trials, emergency approval, and side effects |
+| Public Health Policies | Vaccine mandates, vaccine passports, and/or other public health policies such as social distancing, quarantine, lockdowns, and mask mandates |
+| Vaccination Rollout/Campaign | Vaccination rollout, campaigns, priority groups, vaccine deals, and/or the monitoring of vaccination rates |
+| Vaccine Hesitancy and Mis-/Disinformation | The circulation of disinformation and misinformation regarding vaccines, which contribute to vaccine hesitancy and reluctance |
+| Public Endorsement to Vaccine | Public endorsement and incentive for vaccines, such as celebrities and artists getting vaccinated |
+| Institutional Affairs | Institutional and governmental issues, political disputes, conflicts, political authorities, and export bans |
+| Problems in Vaccination | Problems in vaccination, such as delays, fraud, disparities, vaccine shortages |
+| Public Perception of Vaccine | Opinion polls and surveys on the public perception of vaccines and/or willingness to get vaccinated |
+| Economic Consequences | The impacts and benefits of vaccination on the economy |
+
+## How it works
+
+For each headline, the classifier tests it against every class description as a text-hypothesis pair, following the entailment framework: the headline is the *text* (T), and each class description is a candidate *hypothesis* (H). The class whose description achieves the highest entailment score is selected as the prediction.
+
+## Citation
+
+If you use this library or the underlying model in your research, please cite:
+
+```bibtex
+@article{neves2025boosting,
+  title={Boosting Transformers: Recognizing Textual Entailment for Classification of Vaccine News Coverage},
+  author={Neves, Luiz F. F. and Camargo, Chico Q. and Massarani, Luisa},
+  journal={Computational Communication Research},
+  volume={7},
+  number={1},
+  pages={1--32},
+  year={2025},
+  doi={10.5117/CCR2025.1.1.NEVE}
+}
+```
+
+## License
+
+MIT

classivax-0.1.0/README.md

@@ -0,0 +1,97 @@
+# ClassiVax
+
+Classify vaccine-related news headlines into topic categories using a **Recognizing Textual Entailment (RTE)** approach, powered by the [`LuizNeves/BERT-NLI-vaccine-v2`](https://huggingface.co/LuizNeves/BERT-NLI-vaccine-v2) model.
+
+Built on the method described in:
+
+> Neves, L. F. F., Camargo, C. Q., & Massarani, L. (2025). Boosting Transformers: Recognizing Textual Entailment for Classification of Vaccine News Coverage. *Computational Communication Research*, 7(1), 1–32. https://doi.org/10.5117/CCR2025.1.1.NEVE
+
+## Installation
+
+```bash
+pip install classivax
+```
+
+## Usage
+
+### Default classes (10 vaccine-related topics from the paper)
+
+```python
+from classivax import ClassiVaxClassifier
+
+clf = ClassiVaxClassifier()
+
+result = clf.classify("Twitter will begin removing vaccine misinformation")
+print(result["class"])   # "Vaccine Hesitancy and Mis-/Disinformation"
+print(result["score"])   # e.g. 0.999
+```
+
+### Batch classification
+
+```python
+headlines = [
+    "Twitter will begin removing vaccine misinformation",
+    "Britain begins coronavirus vaccine rollout",
+    "COVID-19 vaccination key to economic recovery",
+]
+
+df = clf.classify_batch(headlines)
+print(df)
+```
+
+### Custom classes
+
+The underlying RTE approach also supports classifying text against any set of natural-language class descriptions you define — not just the original 10 classes.
+
+```python
+custom_classes = {
+    "Vaccine Safety Debate": "The headline discusses concerns, controversies, or debates about vaccine safety",
+    "Vaccine Logistics": "The headline discusses shipping, storage, cold chain, or distribution logistics for vaccines",
+}
+
+clf = ClassiVaxClassifier(classes=custom_classes)
+result = clf.classify("Cold storage shortages delay vaccine shipments to rural areas")
+print(result["class"])  # "Vaccine Logistics"
+```
+
+**Note:** the underlying model was fine-tuned specifically on English-language, vaccine-related headlines from 2020–2021. Using custom classes outside this domain is supported, but accuracy has not been formally evaluated beyond the original study's scope — treat results on very different domains with appropriate caution.
+
+## The 10 default classes
+
+| Class | Description |
+|---|---|
+| Global Access to Vaccine | Initiatives for equal access to COVID-19 vaccines (e.g., COVAX), and/or the need to combat inequity, ensure global distribution, and contain the disease, especially in low- and middle-income countries |
+| Science and Technology | The science behind vaccine development, including research, studies, safety and efficacy trials, emergency approval, and side effects |
+| Public Health Policies | Vaccine mandates, vaccine passports, and/or other public health policies such as social distancing, quarantine, lockdowns, and mask mandates |
+| Vaccination Rollout/Campaign | Vaccination rollout, campaigns, priority groups, vaccine deals, and/or the monitoring of vaccination rates |
+| Vaccine Hesitancy and Mis-/Disinformation | The circulation of disinformation and misinformation regarding vaccines, which contribute to vaccine hesitancy and reluctance |
+| Public Endorsement to Vaccine | Public endorsement and incentive for vaccines, such as celebrities and artists getting vaccinated |
+| Institutional Affairs | Institutional and governmental issues, political disputes, conflicts, political authorities, and export bans |
+| Problems in Vaccination | Problems in vaccination, such as delays, fraud, disparities, vaccine shortages |
+| Public Perception of Vaccine | Opinion polls and surveys on the public perception of vaccines and/or willingness to get vaccinated |
+| Economic Consequences | The impacts and benefits of vaccination on the economy |
+
+## How it works
+
+For each headline, the classifier tests it against every class description as a text-hypothesis pair, following the entailment framework: the headline is the *text* (T), and each class description is a candidate *hypothesis* (H). The class whose description achieves the highest entailment score is selected as the prediction.
+
+## Citation
+
+If you use this library or the underlying model in your research, please cite:
+
+```bibtex
+@article{neves2025boosting,
+  title={Boosting Transformers: Recognizing Textual Entailment for Classification of Vaccine News Coverage},
+  author={Neves, Luiz F. F. and Camargo, Chico Q. and Massarani, Luisa},
+  journal={Computational Communication Research},
+  volume={7},
+  number={1},
+  pages={1--32},
+  year={2025},
+  doi={10.5117/CCR2025.1.1.NEVE}
+}
+```
+
+## License
+
+MIT

classivax-0.1.0/classivax/__init__.py

@@ -0,0 +1,7 @@
+"""ClassiVax — classify vaccine-related news headlines using textual entailment."""
+
+from .classifier import ClassiVaxClassifier
+from .classes_default import DEFAULT_CLASSES
+
+__all__ = ["ClassiVaxClassifier", "DEFAULT_CLASSES"]
+__version__ = "0.1.0"

classivax-0.1.0/classivax/classes_default.py

@@ -0,0 +1,62 @@
+"""
+Default topic classes for vaccine-related news headline classification.
+
+These 10 classes and their descriptions were defined and validated in:
+
+    Neves, L. F. F., Camargo, C. Q., & Massarani, L. (2025). Boosting
+    Transformers: Recognizing Textual Entailment for Classification of
+    Vaccine News Coverage. Computational Communication Research, 7(1), 1-32.
+    https://doi.org/10.5117/CCR2025.1.1.NEVE
+
+See Table 3 of the paper for the full description and illustrative
+headlines for each class.
+"""
+
+DEFAULT_CLASSES = {
+    "Global Access to Vaccine": (
+        "The headline addresses initiatives for equal access to covid-19 "
+        "vaccines, such as COVAX, and/or the need to combat inequity, "
+        "ensure global distribution, and contain the disease, especially "
+        "to low and middle-income countries"
+    ),
+    "Science and Technology": (
+        "The headline addresses the science behind vaccine development, "
+        "including research, studies, safety and efficacy trials, "
+        "emergency approval, and side effects"
+    ),
+    "Public Health Policies": (
+        "The headline addresses vaccine mandates, vaccine passports, "
+        "and/or other public health policies such as social distancing, "
+        "quarantine, lockdowns, and mask mandates"
+    ),
+    "Vaccination Rollout/Campaign": (
+        "The headline addresses vaccination rollout, campaigns, priority "
+        "groups, vaccine deals, and/or the monitoring of vaccination rates"
+    ),
+    "Vaccine Hesitancy and Mis-/Disinformation": (
+        "The headline addresses the circulation of disinformation and "
+        "misinformation regarding vaccines, which contribute to vaccine "
+        "hesitancy and reluctance"
+    ),
+    "Public Endorsement to Vaccine": (
+        "The headline addresses public endorsement and incentive for "
+        "vaccines, such as celebrities and artists getting vaccinated"
+    ),
+    "Institutional Affairs": (
+        "The headline addresses institutional and governmental issues, "
+        "political disputes, conflicts, political authorities, and "
+        "export bans"
+    ),
+    "Problems in Vaccination": (
+        "The headline addresses problems in vaccination, such as delays, "
+        "fraud, disparities, vaccine shortages"
+    ),
+    "Public Perception of Vaccine": (
+        "The headline addresses opinion polls and surveys on the public "
+        "perception of vaccines and/or willingness to get vaccinated"
+    ),
+    "Economic Consequences": (
+        "The headline addresses the impacts and benefits of vaccination "
+        "on the economy"
+    ),
+}

classivax-0.1.0/classivax/classifier.py

@@ -0,0 +1,100 @@
+"""Core classifier for ClassiVax."""
+
+from __future__ import annotations
+
+from typing import Dict, List, Optional
+
+from transformers import pipeline
+
+from .classes_default import DEFAULT_CLASSES
+
+DEFAULT_MODEL = "LuizNeves/BERT-NLI-vaccine-v2"
+
+
+class ClassiVaxClassifier:
+    """Classify news headlines into topic classes using a Recognizing
+    Textual Entailment (RTE) approach.
+
+    By default, this uses the 10 vaccine-related classes described in
+    Neves, Camargo & Massarani (2025) and the fine-tuned model published
+    at https://huggingface.co/LuizNeves/BERT-NLI-vaccine-v2.
+
+    Custom classes can be supplied to adapt the classifier to other
+    domains or research questions. Note that the underlying model was
+    fine-tuned specifically on vaccine-related headlines, so accuracy on
+    very different domains is not guaranteed — see the project README
+    for details.
+
+    Parameters
+    ----------
+    classes:
+        Optional mapping of ``{class_name: natural_language_description}``.
+        If omitted, the 10 default vaccine-related classes are used.
+    model_name:
+        Hugging Face model identifier to load. Defaults to
+        ``LuizNeves/BERT-NLI-vaccine-v2``.
+    """
+
+    def __init__(
+        self,
+        classes: Optional[Dict[str, str]] = None,
+        model_name: str = DEFAULT_MODEL,
+    ) -> None:
+        self.classes = classes if classes is not None else DEFAULT_CLASSES
+        self.model_name = model_name
+        self._pipe = pipeline("text-classification", model=model_name)
+
+    def classify(self, text: str) -> Dict:
+        """Classify a single headline.
+
+        Returns
+        -------
+        dict
+            ``{"class": str, "score": float, "all_scores": dict}``.
+            ``class`` and ``score`` refer to the class with the highest
+            entailment score. ``all_scores`` maps every class name to its
+            raw ``(label, score)`` result, for inspection.
+        """
+        all_results = {}
+        best_class = None
+        best_score = -1.0
+
+        for class_name, description in self.classes.items():
+            result = self._pipe({"text": text, "text_pair": description})
+            all_results[class_name] = {
+                "label": result["label"],
+                "score": result["score"],
+            }
+            entailment_score = (
+                result["score"] if result["label"] == "entailment" else 0.0
+            )
+            if entailment_score > best_score:
+                best_score = entailment_score
+                best_class = class_name
+
+        return {
+            "class": best_class,
+            "score": all_results[best_class]["score"] if best_class else None,
+            "all_scores": all_results,
+        }
+
+    def classify_batch(self, texts: List[str]):
+        """Classify a list of headlines.
+
+        Returns
+        -------
+        pandas.DataFrame
+            One row per input text, with columns ``text``, ``class``, and
+            ``score``.
+        """
+        import pandas as pd
+
+        rows = []
+        for text in texts:
+            result = self.classify(text)
+            rows.append({
+                "text": text,
+                "class": result["class"],
+                "score": result["score"],
+            })
+        return pd.DataFrame(rows)

classivax-0.1.0/classivax.egg-info/PKG-INFO

@@ -0,0 +1,121 @@
+Metadata-Version: 2.4
+Name: classivax
+Version: 0.1.0
+Summary: Classify vaccine-related news headlines into topic categories using a Recognizing Textual Entailment (RTE) approach.
+Author: Luiz Felipe Fernandes Neves
+License: MIT
+Project-URL: Homepage, https://github.com/lffernandes08/classivax
+Project-URL: Repository, https://github.com/lffernandes08/classivax
+Project-URL: Model, https://huggingface.co/LuizNeves/BERT-NLI-vaccine-v2
+Project-URL: Paper, https://doi.org/10.5117/CCR2025.1.1.NEVE
+Keywords: nlp,text-classification,natural-language-inference,textual-entailment,vaccine,science-communication,computational-social-science
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: transformers>=4.30
+Requires-Dist: torch>=2.0
+Requires-Dist: pandas>=1.5
+Dynamic: license-file
+
+# ClassiVax
+
+Classify vaccine-related news headlines into topic categories using a **Recognizing Textual Entailment (RTE)** approach, powered by the [`LuizNeves/BERT-NLI-vaccine-v2`](https://huggingface.co/LuizNeves/BERT-NLI-vaccine-v2) model.
+
+Built on the method described in:
+
+> Neves, L. F. F., Camargo, C. Q., & Massarani, L. (2025). Boosting Transformers: Recognizing Textual Entailment for Classification of Vaccine News Coverage. *Computational Communication Research*, 7(1), 1–32. https://doi.org/10.5117/CCR2025.1.1.NEVE
+
+## Installation
+
+```bash
+pip install classivax
+```
+
+## Usage
+
+### Default classes (10 vaccine-related topics from the paper)
+
+```python
+from classivax import ClassiVaxClassifier
+
+clf = ClassiVaxClassifier()
+
+result = clf.classify("Twitter will begin removing vaccine misinformation")
+print(result["class"])   # "Vaccine Hesitancy and Mis-/Disinformation"
+print(result["score"])   # e.g. 0.999
+```
+
+### Batch classification
+
+```python
+headlines = [
+    "Twitter will begin removing vaccine misinformation",
+    "Britain begins coronavirus vaccine rollout",
+    "COVID-19 vaccination key to economic recovery",
+]
+
+df = clf.classify_batch(headlines)
+print(df)
+```
+
+### Custom classes
+
+The underlying RTE approach also supports classifying text against any set of natural-language class descriptions you define — not just the original 10 classes.
+
+```python
+custom_classes = {
+    "Vaccine Safety Debate": "The headline discusses concerns, controversies, or debates about vaccine safety",
+    "Vaccine Logistics": "The headline discusses shipping, storage, cold chain, or distribution logistics for vaccines",
+}
+
+clf = ClassiVaxClassifier(classes=custom_classes)
+result = clf.classify("Cold storage shortages delay vaccine shipments to rural areas")
+print(result["class"])  # "Vaccine Logistics"
+```
+
+**Note:** the underlying model was fine-tuned specifically on English-language, vaccine-related headlines from 2020–2021. Using custom classes outside this domain is supported, but accuracy has not been formally evaluated beyond the original study's scope — treat results on very different domains with appropriate caution.
+
+## The 10 default classes
+
+| Class | Description |
+|---|---|
+| Global Access to Vaccine | Initiatives for equal access to COVID-19 vaccines (e.g., COVAX), and/or the need to combat inequity, ensure global distribution, and contain the disease, especially in low- and middle-income countries |
+| Science and Technology | The science behind vaccine development, including research, studies, safety and efficacy trials, emergency approval, and side effects |
+| Public Health Policies | Vaccine mandates, vaccine passports, and/or other public health policies such as social distancing, quarantine, lockdowns, and mask mandates |
+| Vaccination Rollout/Campaign | Vaccination rollout, campaigns, priority groups, vaccine deals, and/or the monitoring of vaccination rates |
+| Vaccine Hesitancy and Mis-/Disinformation | The circulation of disinformation and misinformation regarding vaccines, which contribute to vaccine hesitancy and reluctance |
+| Public Endorsement to Vaccine | Public endorsement and incentive for vaccines, such as celebrities and artists getting vaccinated |
+| Institutional Affairs | Institutional and governmental issues, political disputes, conflicts, political authorities, and export bans |
+| Problems in Vaccination | Problems in vaccination, such as delays, fraud, disparities, vaccine shortages |
+| Public Perception of Vaccine | Opinion polls and surveys on the public perception of vaccines and/or willingness to get vaccinated |
+| Economic Consequences | The impacts and benefits of vaccination on the economy |
+
+## How it works
+
+For each headline, the classifier tests it against every class description as a text-hypothesis pair, following the entailment framework: the headline is the *text* (T), and each class description is a candidate *hypothesis* (H). The class whose description achieves the highest entailment score is selected as the prediction.
+
+## Citation
+
+If you use this library or the underlying model in your research, please cite:
+
+```bibtex
+@article{neves2025boosting,
+  title={Boosting Transformers: Recognizing Textual Entailment for Classification of Vaccine News Coverage},
+  author={Neves, Luiz F. F. and Camargo, Chico Q. and Massarani, Luisa},
+  journal={Computational Communication Research},
+  volume={7},
+  number={1},
+  pages={1--32},
+  year={2025},
+  doi={10.5117/CCR2025.1.1.NEVE}
+}
+```
+
+## License
+
+MIT

classivax-0.1.0/classivax.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+LICENSE
+README.md
+pyproject.toml
+classivax/__init__.py
+classivax/classes_default.py
+classivax/classifier.py
+classivax.egg-info/PKG-INFO
+classivax.egg-info/SOURCES.txt
+classivax.egg-info/dependency_links.txt
+classivax.egg-info/requires.txt
+classivax.egg-info/top_level.txt
+tests/test_classifier.py

classivax-0.1.0/classivax.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

classivax-0.1.0/classivax.egg-info/requires.txt

@@ -0,0 +1,3 @@
+transformers>=4.30
+torch>=2.0
+pandas>=1.5

classivax-0.1.0/classivax.egg-info/top_level.txt

@@ -0,0 +1 @@
+classivax

classivax-0.1.0/pyproject.toml

@@ -0,0 +1,44 @@
+[build-system]
+requires = ["setuptools>=68.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "classivax"
+version = "0.1.0"
+description = "Classify vaccine-related news headlines into topic categories using a Recognizing Textual Entailment (RTE) approach."
+readme = "README.md"
+requires-python = ">=3.9"
+license = { text = "MIT" }
+authors = [
+    { name = "Luiz Felipe Fernandes Neves" }
+]
+keywords = [
+    "nlp",
+    "text-classification",
+    "natural-language-inference",
+    "textual-entailment",
+    "vaccine",
+    "science-communication",
+    "computational-social-science",
+]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+    "Intended Audience :: Science/Research",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+]
+dependencies = [
+    "transformers>=4.30",
+    "torch>=2.0",
+    "pandas>=1.5",
+]
+
+[project.urls]
+Homepage = "https://github.com/lffernandes08/classivax"
+Repository = "https://github.com/lffernandes08/classivax"
+Model = "https://huggingface.co/LuizNeves/BERT-NLI-vaccine-v2"
+Paper = "https://doi.org/10.5117/CCR2025.1.1.NEVE"
+
+[tool.setuptools.packages.find]
+include = ["classivax*"]

classivax-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

classivax-0.1.0/tests/test_classifier.py

@@ -0,0 +1,56 @@
+"""
+Basic tests for ClassiVaxClassifier.
+
+These tests download the model from the Hugging Face Hub on first run,
+so they require an internet connection and may take a while the first
+time they execute.
+"""
+
+import pandas as pd
+import pytest
+
+from classivax import ClassiVaxClassifier, DEFAULT_CLASSES
+
+
+@pytest.fixture(scope="module")
+def classifier():
+    return ClassiVaxClassifier()
+
+
+def test_default_classes_loaded(classifier):
+    assert classifier.classes == DEFAULT_CLASSES
+    assert len(classifier.classes) == 10
+
+
+def test_classify_single_headline(classifier):
+    headline = "Twitter will begin removing vaccine misinformation"
+    result = classifier.classify(headline)
+
+    assert result["class"] == "Vaccine Hesitancy and Mis-/Disinformation"
+    assert result["score"] > 0.9
+    assert len(result["all_scores"]) == 10
+
+
+def test_classify_batch(classifier):
+    headlines = [
+        "Twitter will begin removing vaccine misinformation",
+        "Britain begins coronavirus vaccine rollout",
+    ]
+    df = classifier.classify_batch(headlines)
+
+    assert isinstance(df, pd.DataFrame)
+    assert len(df) == 2
+    assert list(df.columns) == ["text", "class", "score"]
+
+
+def test_custom_classes():
+    custom_classes = {
+        "Sports": "The headline is about a sports event or athlete",
+        "Weather": "The headline is about weather conditions or forecasts",
+    }
+    clf = ClassiVaxClassifier(classes=custom_classes)
+
+    assert clf.classes == custom_classes
+
+    result = clf.classify("Storm expected to hit the coast this weekend")
+    assert result["class"] in custom_classes