tmcgirr/BERT-squad-chatbot-AAI

Model Card for Model ID

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

Model Details

Model Description

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]

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

Uses

Basic QA for questions from the SQuAD Dataset

Out-of-Scope Use

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

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

Training Procedure

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

Evaluation

Will be provided soon.