WhatTheMike/Fox-V3
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Don't print everything, use log

Browse Source
This commit is contained in:
bobloy 2021-03-25 09:54:14 -04:00
parent ac9cf1e589
commit b4f20dd7d2
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
chatter/trainers.py
View File

@ -136,7 +136,7 @@ class MovieTrainer(KaggleTrainer):
# log.exception(f"Bad line: {row}") # log.exception(f"Bad line: {row}")
# pass # pass
# else: # else:
# # print(f"Good line: {row}") # # log.info(f"Good line: {row}")
# pass # pass
# #
# # lines_dict = {row[0].strip('"'): row[4] for row in reader_list} # # lines_dict = {row[0].strip('"'): row[4] for row in reader_list}
@ -168,11 +168,10 @@ class MovieTrainer(KaggleTrainer):
statements_from_file.append(statement) statements_from_file.append(statement)
if statements_from_file: if statements_from_file:
print(statements_from_file)
self.chatbot.storage.create_many(statements_from_file) self.chatbot.storage.create_many(statements_from_file)
statements_from_file = [] statements_from_file = []
print("Training took", time.time() - start_time, "seconds.") log.info("Training took", time.time() - start_time, "seconds.")
async def asynctrain(self, *args, **kwargs): async def asynctrain(self, *args, **kwargs):
extracted_lines = self.data_directory / "movie_lines.tsv" extracted_lines = self.data_directory / "movie_lines.tsv"
@ -295,7 +294,7 @@ class UbuntuCorpusTrainer2(KaggleTrainer):
if statements_from_file: if statements_from_file:
self.chatbot.storage.create_many(statements_from_file) self.chatbot.storage.create_many(statements_from_file)
print("Training took", time.time() - start_time, "seconds.") log.info("Training took", time.time() - start_time, "seconds.")
class TwitterCorpusTrainer(Trainer): class TwitterCorpusTrainer(Trainer):