CoolFace
Apppublic

Amrit74/chaybot-nlu

sourceHugging Facemitupdated 11mo agoView on Hugging Face
0likes
App README

๐Ÿค– Chatbot NLU Trainer & Evaluator

A comprehensive platform for training, evaluating, and managing NLU (Natural Language Understanding) models for chatbots.

Features

๐ŸŽฏ Core Features

  • โ€”Intent Classification Training - Train models to understand user intentions
  • โ€”Entity Recognition - Extract key information from user messages
  • โ€”Multi-Backend Support - Train with HuggingFace, Rasa, or spaCy
  • โ€”Model Evaluation - Comprehensive metrics and confusion matrices
  • โ€”Active Learning - Improve models with uncertain predictions
  • โ€”Model Versioning - Track and manage different model versions

๐Ÿ“Š Analytics & Monitoring

  • โ€”Real-time training progress
  • โ€”Performance metrics visualization
  • โ€”Confidence score analysis
  • โ€”Intent distribution charts

๐Ÿ”ง Built With

  • โ€”Frontend: Gradio for interactive UI
  • โ€”Backend: Python with scikit-learn, transformers
  • โ€”Visualization: Plotly for charts and graphs
  • โ€”Storage: JSON-based data management

How to Use

1. Training Tab

  • โ€”Upload your training data (JSON format)
  • โ€”Select backend (HuggingFace/Rasa/spaCy)
  • โ€”Configure training parameters
  • โ€”Start training and monitor progress

2. Evaluation Tab

  • โ€”Test your trained model
  • โ€”View performance metrics
  • โ€”Analyze confusion matrix
  • โ€”Check per-intent statistics

3. Prediction Tab

  • โ€”Enter text to classify
  • โ€”View predicted intent and confidence
  • โ€”See alternative predictions
  • โ€”Get entity extraction results

4. Active Learning

  • โ€”Review uncertain predictions
  • โ€”Provide correct labels
  • โ€”Retrain model with feedback
  • โ€”Improve model accuracy

Sample Data Format

json
[
  {
    "text": "I want to book a flight to New York",
    "intent": "book_flight",
    "entities": [
      {"entity": "destination", "value": "New York"}
    ]
  },
  {
    "text": "Cancel my reservation",
    "intent": "cancel_booking",
    "entities": []
  }
]

Links

Author

Amarjit Kumar

License

MIT License - See LICENSE file for details


This is a demo version optimized for Hugging Face Spaces. For the full-featured application with MongoDB integration, user management, and advanced features, check out the GitHub repository.