changed __repr__

.idea/politweet.iml

@@ -3,7 +3,6 @@
   <component name="NewModuleRootManager">
     <content url="file://$MODULE_DIR$">
       <excludeFolder url="file://$MODULE_DIR$/politweet-environment" />
-      <excludeFolder url="file://$MODULE_DIR$/venv" />
     </content>
     <orderEntry type="inheritedJdk" />
     <orderEntry type="sourceFolder" forTests="false" />

textclassifier/TextClassifier.py

@@ -184,14 +184,14 @@ class TextClassifier:
                          "\nTransportation,\nWorld.\nTOPIC="
 
         response = openai.Completion.create(
-                model=self.model_name,
-                prompt=prompt_string,
-                temperature=0,
-                max_tokens=892,
-                top_p=1,
-                frequency_penalty=0,
-                presence_penalty=0,
-            )
+            model=self.model_name,
+            prompt=prompt_string,
+            temperature=0,
+            max_tokens=892,
+            top_p=1,
+            frequency_penalty=0,
+            presence_penalty=0,
+        )
         classification_unclean = response.choices[0]['text']
         classification_clean = self.cleanup_topic_results(classification_unclean)
 
@@ -207,14 +207,16 @@ class TextClassifier:
 
     @staticmethod
     def cleanup_topic_results(text):
-            new_item = text.replace("\n", " ")
-            new_item = new_item.replace("  ", " ")
-            return new_item
+        new_item = text.replace("\n", " ")
+        new_item = new_item.replace("  ", " ")
+        return new_item
 
     def __repr__(self):
-        return "TwitterScraper(from_date={}, to_date={}, num_tweets={})".format(self.from_date, self.to_date,
-                                                                                self.num_tweets)
-
+        """
+        Gives a string that describes which user is classified
+        :return:
+        """
+        return "Classifier for user: " + self.user_name + " with model: " + self.model_name + "."
 
 
 # if __name__ == "__main__":
@@ -232,4 +234,3 @@ class TextClassifier:
 #     print(df)
 #     # save to csv in a folder under politweet with timestamp in name
 #     df.to_csv(f"{datetime.now().strftime('%Y-%m-%d %H-%M-%S')}_tweets.csv")
-