Update refine_paraphrases.py
Browse files- refine_paraphrases.py +2 -2
refine_paraphrases.py
CHANGED
@@ -5,8 +5,8 @@ from metrics import compute_reward_scores
|
|
5 |
from model_loader import paraphraser_model
|
6 |
|
7 |
# Configuration
|
8 |
-
DATA_PATH = "refined_paraphrases.csv"
|
9 |
-
OUTPUT_PATH = "iterated_paraphrases.csv"
|
10 |
MAX_ITERATIONS = 3
|
11 |
TARGET_SCORES = {
|
12 |
"empathy": 0.9,
|
|
|
5 |
from model_loader import paraphraser_model
|
6 |
|
7 |
# Configuration
|
8 |
+
DATA_PATH = "toxic-comment-classifier_rlhf/refined_paraphrases.csv"
|
9 |
+
OUTPUT_PATH = "toxic-comment-classifier_rlhf/iterated_paraphrases.csv"
|
10 |
MAX_ITERATIONS = 3
|
11 |
TARGET_SCORES = {
|
12 |
"empathy": 0.9,
|