Instructions to use mamakos/CMClassifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use mamakos/CMClassifier with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("teknium/OpenHermes-2.5-Mistral-7B") model = PeftModel.from_pretrained(base_model, "mamakos/CMClassifier") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -32,10 +32,11 @@ import closedmindedness
|
|
| 32 |
model = closedmindedness.classifier()
|
| 33 |
|
| 34 |
text_1 = "Nothing can make me change my mind."
|
| 35 |
-
text_2 = "
|
| 36 |
-
|
|
|
|
| 37 |
probs = model.predict(texts)
|
| 38 |
-
print(probs) # [9.71055567e-01 3.33786011e-06]
|
| 39 |
|
| 40 |
prob = model.predict(text_1)
|
| 41 |
print(prob) # [0.97105557]
|
|
|
|
| 32 |
model = closedmindedness.classifier()
|
| 33 |
|
| 34 |
text_1 = "Nothing can make me change my mind."
|
| 35 |
+
text_2 = "Let's discuss this."
|
| 36 |
+
text_3 = "I went to the movies yesterday."
|
| 37 |
+
texts = [text_1, text_2, text_3]
|
| 38 |
probs = model.predict(texts)
|
| 39 |
+
print(probs) # [9.71055567e-01 1.70469284e-05 3.33786011e-06]
|
| 40 |
|
| 41 |
prob = model.predict(text_1)
|
| 42 |
print(prob) # [0.97105557]
|