You can deploy this model and start earning money today!

alexyang790/
gpt2-classbot

Name: gpt2-classbot
Availability: InStock

text-generation-inference

endpoints_compatible

Views

No views yet

language: en tags:

gpt2
text-generation
class-scheduling
fine-tuned license: mit datasets: web-scrapped dataset from Louslist.org with a custom API model-index:
name: GPT-2 UVA Class Scheduler

🧠 GPT-2 UVA Class Scheduler (Fine-Tuned)

This model is a fine-tuned version of gpt2 on a custom dataset extracted from Lou's List — the University of Virginia's course directory. It is designed to answer questions about course offerings, availability, instructors, and generate plausible class schedules in natural language.

🔍 Use Cases

Ask for available classes in a subject
Query instructors teaching specific courses
Get summaries of course descriptions
Generate sample schedules for a semester

🚀 Example Usage

python

from transformers import GPT2LMHeadModel, GPT2Tokenizer

model = GPT2LMHeadModel.from_pretrained("your-username/uva-gpt2-scheduler")
tokenizer = GPT2Tokenizer.from_pretrained("your-username/uva-gpt2-scheduler")

prompt = "What classes are available for Data Science in Spring 2024?"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_length=100)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))

🏋️ Training

Base model: gpt2 (117M parameters)
Dataset: Custom dataset scraped and cleaned from UVA's Lou’s List
Training platform: Google Colab
Epochs: 3
Batch size: 4
Learning rate: 5e-5
Tokenizer: GPT-2 tokenizer
Token Length: 128

📁 Model Files

pytorch_model.bin: Model weights
config.json: Model configuration
tokenizer.json, vocab.json, merges.txt: Tokenizer files

🧪 Limitations

The model is trained only on course listing data, so it may hallucinate responses if prompted with unrelated queries.
Class availability and schedules may change over time — this model reflects a snapshot of the data at the time of training.

📜 License

MIT License

🙏 Acknowledgements

Course data provided via Lou’s List (https://louslist.org)
Fine-tuning done for the [Class Project Name] at the University of Virginia M.S. in Data Science

alexyang790/gpt2-classbot