boltuix commited on
Commit
29f788b
Β·
verified Β·
1 Parent(s): ac57741

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +52 -3
README.md CHANGED
@@ -16,11 +16,28 @@ pipeline_tag: text-classification
16
  library_name: transformers
17
  tags:
18
  - text-classification
19
- - trip
20
  - multi-text-classification
21
  - classification
 
 
 
 
 
 
 
 
 
 
 
 
 
 
22
  ---
23
 
 
 
 
 
24
  # 🌍 NeuroLocale β€” Your Smarter Nearby Assistant! πŸ—ΊοΈ
25
 
26
  [![License: Open Source](https://img.shields.io/badge/License-Open%20Source-green.svg)](https://opensource.org/licenses)
@@ -91,6 +108,7 @@ print(result) # πŸ–¨οΈ Example Output: [{'label': 'game', 'score': 0.9999}]
91
 
92
  ---
93
 
 
94
  ## Supported Categories πŸͺ
95
 
96
  NeuroLocale supports **120+ local business categories**, each paired with an emoji for clarity:
@@ -228,7 +246,7 @@ Get started with NeuroLocale:
228
  pip install transformers torch pandas scikit-learn tqdm
229
  ```
230
 
231
- - **Requirements** πŸ“‹: Python 3.8+, ~500MB storage for model and dependencies.
232
  - **Optional** πŸ”§: CUDA-enabled GPU for faster training/inference.
233
  - **Model Download** πŸ“₯: Grab the pre-trained model from [Hugging Face](https://huggingface.co/boltuix/NeuroLocale).
234
 
@@ -247,6 +265,37 @@ NeuroLocale is trained using **NeuroBERT** for multi-class text classification.
247
  ...
248
  ```
249
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
250
  ### Training Code
251
  ```python
252
  import pandas as pd
@@ -491,7 +540,7 @@ NeuroLocale was tested on **122 test cases**, achieving **94.26% accuracy** (115
491
  - **Source**: Open-source datasets, augmented with custom queries (e.g., ChatGPT, Grok, or proprietary data).
492
  - **Format**: CSV with `text` (query) and `label` (category) columns.
493
  - **Categories**: 120+ (see [Supported Categories](#supported-categories)).
494
- - **Size**: Varies based on dataset; model footprint ~500MB.
495
  - **Preprocessing**: Handled via tokenization and label encoding (see [Training the Model](#training-the-model)).
496
 
497
  ---
 
16
  library_name: transformers
17
  tags:
18
  - text-classification
 
19
  - multi-text-classification
20
  - classification
21
+ - intent-classification
22
+ - intent-detection
23
+ - nlp
24
+ - natural-language-processing
25
+ - transformers
26
+ - edge-ai
27
+ - iot
28
+ - smart-home
29
+ - location-intelligence
30
+ - voice-assistant
31
+ - conversational-ai
32
+ - real-time
33
+ - boltuix
34
+ - neurobert
35
  ---
36
 
37
+
38
+ ![Banner](https://blogger.googleusercontent.com/img/b/R29vZ2xl/AVvXsEg0DZmojOOjUG9wZGTDZ4xIkVO6ifyCF06eO4N4x3cKIDPRZlym5tlCP3qfJJSK7UVFix2TUvzvrJDPFyuylmzJrPiCWxwWou74YXhSwMhRVohlG34w9Ln29MbhzU_NSc1rRzfZgJE8i7BViLiuSR7hyphenhyphenEBYg12C26HCyLne5VLkWzfClbMRKAGzMJSxv-4/s6250/NeuroLocale.jpg)
39
+
40
+
41
  # 🌍 NeuroLocale β€” Your Smarter Nearby Assistant! πŸ—ΊοΈ
42
 
43
  [![License: Open Source](https://img.shields.io/badge/License-Open%20Source-green.svg)](https://opensource.org/licenses)
 
108
 
109
  ---
110
 
111
+
112
  ## Supported Categories πŸͺ
113
 
114
  NeuroLocale supports **120+ local business categories**, each paired with an emoji for clarity:
 
246
  pip install transformers torch pandas scikit-learn tqdm
247
  ```
248
 
249
+ - **Requirements** πŸ“‹: Python 3.8+, ~50MB storage for model and dependencies.
250
  - **Optional** πŸ”§: CUDA-enabled GPU for faster training/inference.
251
  - **Model Download** πŸ“₯: Grab the pre-trained model from [Hugging Face](https://huggingface.co/boltuix/NeuroLocale).
252
 
 
265
  ...
266
  ```
267
 
268
+
269
+ # πŸ€– Supported Categories from `boltuix/NeuroLocale`
270
+
271
+ This file shows how to extract the full list of intent labels supported by the `boltuix/NeuroLocale` model using Hugging Face Transformers.
272
+
273
+ ---
274
+
275
+ ## πŸ”§ How to List All Supported Categories
276
+
277
+ ```python
278
+ from transformers import AutoModelForSequenceClassification
279
+
280
+ # πŸ“₯ Load the fine-tuned intent classification model
281
+ model = AutoModelForSequenceClassification.from_pretrained("boltuix/NeuroLocale")
282
+
283
+ # 🏷️ Extract the ID-to-label mapping dictionary
284
+ label_mapping = model.config.id2label
285
+
286
+ # πŸ“‹ Convert and sort all labels to a clean list
287
+ supported_labels = sorted(label_mapping.values())
288
+
289
+ # βœ… Print the supported categories
290
+ print("βœ… Supported Categories:", supported_labels)
291
+
292
+ #βœ… Output
293
+ #βœ… Supported Categories: ['accounting firm', 'airport', 'amusement park', ',...
294
+
295
+ ```
296
+ ---
297
+
298
+
299
  ### Training Code
300
  ```python
301
  import pandas as pd
 
540
  - **Source**: Open-source datasets, augmented with custom queries (e.g., ChatGPT, Grok, or proprietary data).
541
  - **Format**: CSV with `text` (query) and `label` (category) columns.
542
  - **Categories**: 120+ (see [Supported Categories](#supported-categories)).
543
+ - **Size**: Varies based on dataset; model footprint ~50MB.
544
  - **Preprocessing**: Handled via tokenization and label encoding (see [Training the Model](#training-the-model)).
545
 
546
  ---