Upload runCorpus.py

Fixed a lot of errors caused by sublime. They were fixed before but file mismanagement; Glad production updates on hugging face

runCorpus.py

@@ -15,7 +15,7 @@ from preprocessed_dialogs import dialog_data
 
 class CorpusTrainer:
 	def __init__(self):
-		self.running_trouble = None
+		self.runningTrouble = []
 		self.choices_yes = ["yes", "ya", "yeah", "yessir", "yesir", "y", "ye", "yah"]
 		self.exit_commands = ["exit", "quit", "stop", "x", "q", ""]
 
@@ -146,7 +146,7 @@ class CorpusTrainer:
 			with open("troubled_speakers.txt", 'a') as f:
 				f.write(f"{speaker}\n")
 
-		self.allTogether = self.resetTogether(self.speakerList, self.troubleList)
+		self.allTogether = self.resetTogether()
 		topConvo = len(self.allTogether)
 		self.bad_count = len(self.troubleList)
 
@@ -158,7 +158,7 @@ class CorpusTrainer:
 		percent_running = self.runningPercent(len(self.troubleList), self.counter)
 		self.failure_history.append(len(self.troubleList))
 		if percent_running is None:
-			percent_self.running = 0.0
+			percent_running = 0.0
 		self.chatbot_trainer.logger.info(f"self.Running Percentage Failure: {percent_running}%")
 
 		# We check for speaker vs top num achieved successfully in self.speakerList
@@ -169,8 +169,8 @@ class CorpusTrainer:
 
 		if percent_running is not None:
 			# Note: The run adds to the trained speaker list which is successful trainings. If it's not in that list, when it comes up again it will be trained another time.
-			if percent_running > percent_reset:
-				print("Logging Failures... Resetting... Failure Rate is Greater than {percent_reset}%...")
+			if percent_running > self.percent_reset:
+				print("Logging Failures... Resetting... Failure Rate is Greater than {self.percent_reset}%...")
 				answer_1 = input("Show Failures for this Run?	\n>")
 				if answer_1 in self.exit_commands:
 					quit()
@@ -211,7 +211,7 @@ class CorpusTrainer:
 				f.write(f"{speaker}\n")
 
 		# Find Top Convo
-		self.allTogether = self.resetTogether(self.speakerList, self.troubleList)
+		self.allTogether = self.resetTogether()
 		topConvo = len(self.allTogether)
 		self.bad_count = len(self.troubleList)
 
@@ -222,41 +222,42 @@ class CorpusTrainer:
 		print(f"Number of Conversations Combined: {topConvo}")
 		print(f"self.Running Trouble: {len(self.runningTrouble)}")
 
-		percent_self.running = self.runningPercent(len(self.troubleList), topConvo)
+		percent_running = self.runningPercent(len(self.troubleList), topConvo)
 		self.failure_history.append(len(self.troubleList))
-		if percent_self.running is None:
-			percent_self.running = 0.0
-		self.chatbot_trainer.logger.info(f"self.Running Percentage Failure: {percent_self.running}%")
+		if percent_running is None:
+			percent_running = 0.0
+		self.chatbot_trainer.logger.info(f"self.Running Percentage Failure: {percent_running}%")
 
 		print(f"Now is the time to quit if need be...  ")
 		if play_notification == 1:
 			playsound(notification_sound)
-		for x in range(time_sleep):
+		for x in range(self.time_sleep):
 			# time.sleep(1)
-			print(f"Next convo in:{time_sleep-x}")
+			print(f"Next convo in:{self.time_sleep-x}")
 
-		if percent_self.running is not None:
+		if percent_running is not None:
 			# Note: The self.run adds to the trained speaker list which is successful trainings. If it's not in that list, when it comes up again it will be trained another time.
-			if percent_self.running > percent_reset:
+			if percent_running > self.percent_reset:
 				self.log_failures(len(self.troubleList), log_file)
 				print("Plotting Failures... See failures_plot.png for more information...  ")
 				self.plot_failures(log_file)
 				if play_notification == 1:
 					playsound(notification_sound)
-				print(f"Resetting... Failure Rate is Greater than {percent_reset}%...  For this self.run.")
-				# time.sleep(time_sleep * 3)
+				print(f"Resetting... Failure Rate is Greater than {self.percent_reset}%...  For this self.run.")
+				# time.sleep(self.time_sleep * 3)
 				return self.main(self.chatbot_trainer, user_choice, dialog_data, topConvo, self.top_num)
 
-	def resetTogether():
+	def resetTogether(self):
 		for speakers in self.speakerList:
 			if speakers not in self.allTogether:
 				self.allTogether.append(str(speakers))
 				for speakers in self.troubleListData:
 					if speakers not in self.allTogether:
 						self.allTogether.append(str(speakers))
-						self.allTogetherSorted = sorted(self.allTogether)
+				
+		allTogetherSorted = sorted(self.allTogether)
 
-		return self.allTogetherSorted
+		return allTogetherSorted
 
 	def cleanupTrained(self, speakerList):
 		for data in self.speakerList: