Misc changes for comma experiments

curt-tigges · Oct 13, 2023 · a82b82d · a82b82d
1 parent 2bc4766
commit a82b82d
Show file tree

Hide file tree

Showing 9 changed files with 27,510 additions and 1,445 deletions.
diff --git a/circuit_analysis_sentiment_classification_pythia1_4b.ipynb b/circuit_analysis_sentiment_classification_pythia1_4b.ipynb
@@ -16082,18 +16082,6 @@
    "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.9.16"
   }
  },
  "nbformat": 4,

diff --git a/circuit_analysis_sentiment_continuation_pythia1_4b.ipynb b/circuit_analysis_sentiment_continuation_pythia1_4b.ipynb
@@ -52022,18 +52022,6 @@
    "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.9.16"
   }
  },
  "nbformat": 4,

diff --git a/circuit_analysis_simple_sentiment_gpt2_small.ipynb b/circuit_analysis_simple_sentiment_gpt2_small.ipynb
diff --git a/circuit_for_mood_inference_pythia2_8b_commas.ipynb b/circuit_for_mood_inference_pythia2_8b_commas.ipynb
diff --git a/comma_ablation_experiments copy.ipynb b/comma_ablation_experiments copy.ipynb
@@ -5782,18 +5782,6 @@
    "language": "python",
    "name": "python3"
   },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.9.16"
-  },
   "orig_nbformat": 4
  },
  "nbformat": 4,

diff --git a/comma_ablation_experiments.ipynb b/comma_ablation_experiments.ipynb
diff --git a/projection_neuroscope.py b/projection_neuroscope.py
diff --git a/projection_neuroscope2.ipynb b/projection_neuroscope2.ipynb
diff --git a/utils/prompts.py b/utils/prompts.py
@@ -110,6 +110,7 @@ class PromptType(Enum):
     SIMPLE_TEST = "simple_test"
     COMPLETION = "completion"
     COMPLETION_2 = "completion_2"
+    COMPLETION_3 = "completion_3"
     CLASSIFICATION = "classification"
     CLASSIFICATION_2 = "classification_2"
     CLASSIFICATION_3 = "classification_3"
@@ -134,6 +135,7 @@ def get_format_string(self):
             PromptType.MEDICAL: "I heard the doctor use the word{MED} and I felt very",
             PromptType.COMPLETION: "I thought this movie was{ADJ1}, I{VRB} it. The acting was{ADJ2}, the plot was{ADJ3}, and overall the movie was just very",
             PromptType.COMPLETION_2: "I thought this movie was{ADJ1}, I{VRB} it. The acting was{ADJ2}, the plot was{ADJ3}, and overall it was just very",
+            PromptType.COMPLETION_3: "I thought this movie was{ADJ1}, I{VRB} it. The acting was{ADJ2}, the plot was{ADJ3}, and overall the movie was just very {ADJ4}. I felt it was",
             PromptType.CLASSIFICATION: "Review Text: 'I thought this movie was{ADJ1}, I{VRB} it. The acting was{ADJ2}, the plot was{ADJ3}, and overall the movie was just very {ADJ4}.' \nReview Sentiment:",
             PromptType.CLASSIFICATION_2: "Review Text: I thought this movie was{ADJ1}, I{VRB} it. The acting was{ADJ2}, the plot was{ADJ3}, and overall the movie was just very {ADJ4}. \nReview Sentiment:",
             PromptType.CLASSIFICATION_3: "Review Text: I thought this movie was{ADJ1}, I{VRB} it. The acting was{ADJ2}, the plot was{ADJ3}, and overall the movie was just very {ADJ4}. Review Sentiment:",