Model Card for Menstrual Health Awareness Model
This model is designed to answer questions related to menstrual health using a dataset of questions and answers on the topic.
Model Use Guide
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
# Load the tokenizer and model
tokenizer = AutoTokenizer.from_pretrained("adi2606/Menstrual-Health-Awareness-Chatbot")
model = AutoModelForSeq2SeqLM.from_pretrained("adi2606/Menstrual-Health-Awareness-Chatbot")
# Function to generate a response from the chatbot
def generate_response(input_text):
# Tokenize the input text
inputs = tokenizer(input_text, return_tensors="pt")
# Generate the output (the response)
outputs = model.generate(**inputs)
# Decode the output back to text
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
return response
# Example usage
input_text = "What are the symptoms of PMS?"
response = generate_response(input_text)
print(response)
Model Description
This model aims to provide accurate and relevant answers to questions about menstrual health. It has been trained on a dataset that includes a variety of questions and answers covering different aspects of menstrual health, including menstrual cycle length, symptoms, and general health advice.
- Model type: Sequence-to-sequence language model
- Language(s) (NLP): English
- License: MIT
Uses
Direct Use
The model can be used to answer questions related to menstrual health. Users can input their questions, and the model will generate relevant answers based on its training data.
Out-of-Scope Use
The model should not be used for medical diagnosis or treatment recommendations. It is designed for informational purposes and should not replace professional medical advice.
Bias, Risks, and Limitations
The model's responses are limited by the scope and quality of the training data. It may not provide accurate answers to all possible questions about menstrual health, especially those that require personalized medical advice.
Recommendations
Users should consider the model as a supplemental information source and consult healthcare professionals for medical advice. Further evaluation and updates to the training data can help mitigate biases and improve accuracy.
Training Details
Training Data The model was trained on the Menstrual Health Awareness Dataset, which includes questions and answers related to menstrual health.
Training Procedure
Data was tokenized and preprocessed to ensure consistency in sequence length, with appropriate padding and truncation.
Training Hyperparameters Training regime: fp32 mixed precision Learning rate: 2e-5 Batch size: 16 Epochs: 3 Weight decay: 0.01
Evaluation
Testing Data, Factors & Metrics Testing Data The model was evaluated on a test split of the Menstrual Health Awareness Dataset.
Factors
The evaluation considered the accuracy and relevance of the answers generated by the model.
Metrics
Accuracy: The percentage of correct answers generated by the model. BLEU Score: A measure of the similarity between the generated answer and the reference answer. Results
Summary
The model achieved an accuracy of 85% and a BLEU score of 70 on the test dataset, indicating a high level of relevance and correctness in its responses.
Environmental Impact
Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).
Hardware Type: NVIDIA Tesla V100 Cloud Provider: Google Cloud Compute Region: us-central1 Carbon Emitted: 15 kg CO2eq (estimated)
Compute Infrastructure
Hardware GPU: NVIDIA Tesla V100 Software Framework: Transformers library by Hugging Face Python Version: 3.8
- Downloads last month
- 30