{"id":19316738,"url":"https://github.com/vubacktracking/mamba-text-classification","last_synced_at":"2025-04-22T17:30:28.612Z","repository":{"id":236946202,"uuid":"792669750","full_name":"VuBacktracking/mamba-text-classification","owner":"VuBacktracking","description":"Text Classification using Mamba Model","archived":false,"fork":false,"pushed_at":"2024-07-15T06:33:45.000Z","size":23,"stargazers_count":23,"open_issues_count":2,"forks_count":7,"subscribers_count":1,"default_branch":"main","last_synced_at":"2025-04-02T02:02:01.384Z","etag":null,"topics":["deep-learning","imdb-dataset","mamba","text-classification","torch"],"latest_commit_sha":null,"homepage":"","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/VuBacktracking.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2024-04-27T08:22:50.000Z","updated_at":"2025-03-23T05:21:18.000Z","dependencies_parsed_at":"2024-04-29T11:24:35.391Z","dependency_job_id":"dcca5edf-9afc-4995-a982-e635753bb56e","html_url":"https://github.com/VuBacktracking/mamba-text-classification","commit_stats":null,"previous_names":["vubacktracking/mamba-text-classification"],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/VuBacktracking%2Fmamba-text-classification","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/VuBacktracking%2Fmamba-text-classification/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/VuBacktracking%2Fmamba-text-classification/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/VuBacktracking%2Fmamba-text-classification/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/VuBacktracking","download_url":"https://codeload.github.com/VuBacktracking/mamba-text-classification/tar.gz/refs/heads/main","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":250287337,"owners_count":21405588,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["deep-learning","imdb-dataset","mamba","text-classification","torch"],"created_at":"2024-11-10T01:12:29.284Z","updated_at":"2025-04-22T17:30:28.305Z","avatar_url":"https://github.com/VuBacktracking.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"# Text Classification using Mamba with IMDB Dataset\n\n## Overview\nThis project aims to perform sentiment analysis on the IMDB movie review dataset using the Mamba Model. The goal is to classify movie reviews as either positive or negative based on their textual content.\n\n## Dataset\nThe IMDB dataset consists of 50,000 movie reviews, split evenly into 25k for training and 25k for testing. Each review is labeled as either positive or negative.\n\n## Installation\nTo run the project locally, follow these steps:\n\n1. Clone this repository:\n```\ngit clone https://github.com/VuBacktracking/mamba-text-classification.git\n```\n2. Install the required dependencies:\n```\npip install -r requirements.txt\n```\n\n## Usage\n1. Navigate to the project directory:\n```\ncd mamba-text-classification\npython trainer.py\n```\n\n## History of my training\n| Step | Training Loss | Validation Loss | Accuracy |\n|------|---------------|-----------------|----------|\n| 625  | 0.020500      | 0.246246        | 0.928000 |\n| 1250 | 0.671000      | 0.195849        | 0.940800 |\n| 1875 | 0.596100      | 0.266093        | 0.934400 |\n| 2500 | 0.016700      | 0.217099        | 0.941200 |\n| 3125 | 0.000700      | 0.209536        | 0.944800 |\n| 3750 | 2.680700      | 0.188751        | 0.949200 |\n| 4375 | 0.015500      | 0.224948        | 0.950000 |\n| 5000 | 0.002100      | 0.199092        | 0.952800 |\n| 5625 | 0.013400      | 0.192042        | 0.952400 |\n| 6250 | 0.152500      | 0.190083        | 0.953600 |\n\n**Note**: You can check my model on hugging face hub in the link: https://huggingface.co/vubacktracking/mamba_text_classification\n\n## Dependencies\n- Python 3.x\n- Other dependencies listed in requirements.txt\n\n## License\nThis project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.\n\n## Acknowledgments\n- The IMDB dataset: [http://ai.stanford.edu/~amaas/data/sentiment/](http://ai.stanford.edu/~amaas/data/sentiment/)\n- Mamba: [Mamba: Linear-Time Sequence Modeling with Selective State Spaces](https://arxiv.org/pdf/2312.00752)","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fvubacktracking%2Fmamba-text-classification","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fvubacktracking%2Fmamba-text-classification","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fvubacktracking%2Fmamba-text-classification/lists"}