4nkh commited on
Commit
d9d84f8
·
verified ·
1 Parent(s): 36b732d

Create README.md

Browse files

# Theme Model (Multi-Label Classifier)

This model performs **multi-label theme detection** for community knowledge submissions.
It classifies short narratives or summaries into one or more of the following themes:

- **Mentorship**
- **Entrepreneurship**
- **Startup Success**

## 🧩 Base Model
Fine-tuned from [`distilbert-base-uncased`](https://huggingface.co/distilbert-base-uncased).

## 📚 Training Data
Trained on a synthetic dataset (`theme_response.json`) consisting of community stories with labeled themes such as:
- “Mentorship strengthens community-based skill development.”
- “Entrepreneurship drives local economic growth and sustainability.”
- “Support programs help boost startup success rates.”

## ⚙️ Usage
```python
from transformers import pipeline

classifier = pipeline("text-classification", model="4nkh/theme_model", tokenizer="4nkh/theme_model", top_k=None)
text = "Our incubator helped women founders scale their startups."
preds = classifier(text)
print(preds)

Files changed (1) hide show
  1. README.md +35 -0
README.md ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ base_model:
6
+ - google-bert/bert-base-uncased
7
+ pipeline_tag: text-classification
8
+ tags:
9
+ - multi-label
10
+ - theme
11
+ - theme_detection
12
+ - mentorship
13
+ - entrepreneurship
14
+ - startup_success
15
+ - community
16
+ - AI
17
+ - JSON
18
+ - Json_automation
19
+ metrics:
20
+ - f1
21
+ - precision
22
+ - recall
23
+ - accuracy
24
+ library_name: transformers
25
+ ---
26
+ 3---
27
+ license: apache-2.0
28
+ language:
29
+ - en
30
+ base_model:
31
+ - google-bert/bert-base-uncased
32
+ pipeline_tag: text-classification
33
+ tags:
34
+ - multi-label
35
+ ---