BERT-squad-chatbot-AAI

Model Card for Model ID

<!-- Provide a quick summary of what the model is/does. -->

Model trained using BERT ("bert-base-cased") for AAI NLP Course.

Model Details

Model Description

<!-- Provide a longer summary of what this model is. -->

Model trained using BERT ("bert-base-cased") on the SQuAD Dataset.

• Developed by: [Trevor McGirr, Eyoha Mengistu, Reed Oken]
• Model type: [Transformer]
• Language(s) (NLP): [English]
• License: [Unknown]

Model Sources [optional]

<!-- Provide the basic links for the model. -->

• Repository: https://github.com/tmcgirr/SQUAD-BERT-chatbot-AAI
• Demo: https://squad-bert-chatbot-aai.streamlit.app/

Uses

<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->

Basic QA for questions from the SQuAD Dataset

Out-of-Scope Use

<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->

Model may not perform well on current events or uncommon language.

How to Get Started with the Model

Use the Interface API to interact with the model.

Training Details

Training Data

<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->

https://www.kaggle.com/datasets/stanfordu/stanford-question-answering-dataset/data

Training Procedure

<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->

https://github.com/tmcgirr/SQUAD-BERT-chatbot-AAI/blob/master/chatbot_training/chatbot.ipynb

Evaluation

<!-- This section describes the evaluation protocols and provides the results. -->

Will be provided soon.