/model-bert-faqclass

Custom BERT model for F.A.Q. classification with custom keywords

Primary LanguagePythonMIT LicenseMIT

BERT Faq Classification - Machine Learning Model

contributions welcome

Welcome in the repository of the machine learning model for the bert-faqclass project! This repository contains the code to design a machine learning model for faq classification based on BERT, with special handling of keywords that may be out of vocabulary.

Note: check out the use case description in this Medium article.

Installing

It's easy! You just need to install the virtual environment.

python -m venv venv

And then you can activate it and install the requirements.

source ./venv/bin/activate
pip install -r package/requirements.txt

Running the package

You can run the package with the command

python -m bert-faqclass

Authors

  • Nicola Massarenti

License

This project is licensed under the MIT License - see the LICENSE.md file for details.