File size: 1,122 Bytes
ac81450 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | ---
language: en
license: mit
library_name: sklearn
tags:
- text-classification
- sentiment-analysis
- code-review
- sklearn
pipeline_tag: text-classification
---
# Code Review Sentiment Classifier
A lightweight sklearn-based classifier for code review comments. Classifies review feedback as positive, neutral, or negative.
## Model Details
- **Type:** TF-IDF + Logistic Regression pipeline
- **Task:** 3-class text classification
- **Framework:** scikit-learn
- **Labels:** negative (0), neutral (1), positive (2)
## Usage
```python
import pickle
with open("model.pkl", "rb") as f:
model = pickle.load(f)
review = "Great implementation, clean code!"
label = model.predict([review])[0] # 0=negative, 1=neutral, 2=positive
proba = model.predict_proba([review])[0]
```
## Training Data
30 code review comments (10 per class) covering:
- **Positive:** Praise, LGTM, good patterns
- **Neutral:** Suggestions, minor nits, questions
- **Negative:** Bugs, security issues, performance problems
## Limitations
- Small training set
- English only
- Focused on software engineering domain
## License
MIT
|