Update README.md

README.md

@@ -62,51 +62,6 @@ for idx in range(1, len(example["completions"])+1):
 # Step 9  Predicted (score): False (0.97) Label: False
 ```
 
-Example 1)
-
-```python
-from datasets import load_dataset
-from transformers import pipeline
-import os
-os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
-
-model_name = "plaguss/Qwen2.5-0.5B-Math-Shepherd-PRM-0.2"
-
-pipe = pipeline("token-classification", model=model_name, device="cuda")
-dataset = load_dataset("trl-lib/math_shepherd")
-example = dataset["test"][10]
-
-sep = "\n"
-
-print(sep.join((example["prompt"], *example["completions"])))
-for idx in range(1, len(example["completions"])+1):
-    text = sep.join((example["prompt"], *example["completions"][0:idx])) + sep
-    output = pipe(text)
-    score = float(output[-1]["score"])
-    pred = True if output[-1]["entity"] == "LABEL_1" else False
-    print(f"Step {idx}\tPredicted (score): {pred} ({score:.2f})\tLabel: {example['labels'][idx-1]}")
-
-# Grandma gave Bryce and Carter some raisins. Bryce received 6 more raisins than Carter, and Carter received half the number of raisins Bryce received. How many raisins did Bryce receive?
-# Step 1: Let $b$ be the number of raisins Bryce received and $c$ be the number of raisins Carter received.
-# Step 2: We are given that $b = c + 6$ and $c = \frac{1}{2}b$.
-# Step 3: Substituting the second equation into the first equation, we get $b = c + 6 = \frac{1}{2}b + 6$.
-# Step 4: Simplifying, we have $b = \frac{1}{2}b + 6$.
-# Step 5: Subtracting $\frac{1}{2}b$ from both sides, we get $\frac{1}{2}b - b = 6$.
-# Step 6: Simplifying further, we have $\frac{1}{2}b - 2b = 6$.
-# Step 7: Combining like terms, we have $-\frac{1}{2}b = 6$.
-# Step 8: Multiplying both sides by $-2$, we get $b = -12$.
-# Step 9: Therefore, Bryce received $\boxed{-12}$ raisins.The answer is: -12
-# Step 1  Predicted (score): True (0.99)  Label: True
-# Step 2  Predicted (score): True (0.99)  Label: True
-# Step 3  Predicted (score): True (0.94)  Label: True
-# Step 4  Predicted (score): True (0.82)  Label: True
-# Step 5  Predicted (score): True (0.58)  Label: True
-# Step 6  Predicted (score): False (0.62) Label: False
-# Step 7  Predicted (score): False (0.77) Label: False
-# Step 8  Predicted (score): False (0.91) Label: False
-# Step 9  Predicted (score): False (0.97) Label: False
-```
-
 Example 2)
 
 ```python
@@ -193,16 +148,16 @@ for i, example in enumerate(examples):
         pred = True if output[-1]["entity"] == "LABEL_1" else False
         print(f"Step {idx}\tPredicted (score): {pred} ({score:.2f})\tLabel: {example['labels'][idx-1]}")
 
-- Example 0:
-Step 1  Predicted (score): True (0.90)  Label: True
-Step 2  Predicted (score): False (0.55) Label: True
-Step 3  Predicted (score): False (0.62) Label: True
-Step 4  Predicted (score): False (0.90) Label: True
-- Example 1:
-Step 1  Predicted (score): True (0.90)  Label: True
-Step 2  Predicted (score): False (0.55) Label: True
-Step 3  Predicted (score): False (0.62) Label: True
-Step 4  Predicted (score): False (0.96) Label: False
+# - Example 0:
+# Step 1  Predicted (score): True (0.90)  Label: True
+# Step 2  Predicted (score): False (0.55) Label: True
+# Step 3  Predicted (score): False (0.62) Label: True
+# Step 4  Predicted (score): False (0.90) Label: True
+# - Example 1:
+# Step 1  Predicted (score): True (0.90)  Label: True
+# Step 2  Predicted (score): False (0.55) Label: True
+# Step 3  Predicted (score): False (0.62) Label: True
+# Step 4  Predicted (score): False (0.96) Label: False
 ```