Automated document classification system using PyTorch & TorchText. Loads and preprocesses news articles, trains a text classification model, visualizes embeddings, and predicts topics such as World, Sports, Business, and Sci/Tech.
nlp machine-learning natural-language-processing deep-learning text-classification pytorch document-classification ag-news embeddingbag orchtext
-
Updated
Aug 8, 2025 - Jupyter Notebook