init

2022-01-10 15:03:58 +08:00
parent 573689a591
commit 3783ef0e75
57 changed files with 9520 additions and 0 deletions
@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+#############################################################
+# File: Reporter.py
+# Created Date: Tuesday September 24th 2019
+# Author: Chen Xuanhong
+# Email: chenxuanhongzju@outlook.com
+# Last Modified:  Sunday, 4th July 2021 11:50:12 pm
+# Modified By: Chen Xuanhong
+# Copyright (c) 2019 Shanghai Jiao Tong University
+#############################################################
+
+import datetime
+import os
+
+class Reporter:
+    def __init__(self,reportPath):
+        self.path           = reportPath
+        self.withTimeStamp  = False
+        self.index          = 1
+        self.timeStrFormat  = '%Y-%m-%d %H:%M:%S'
+        timeStr = datetime.datetime.strftime(datetime.datetime.now(),'%Y%m%d%H%M%S')
+        self.path = self.path + "-%s.log"%timeStr 
+        if not os.path.exists(self.path):
+            f = open(self.path,'w')
+            f.close()
+    
+    def writeInfo(self,strLine):
+        with open(self.path,'a+') as logf:
+            timeStr = datetime.datetime.strftime(datetime.datetime.now(),self.timeStrFormat)
+            logf.writelines("[%d]-[%s]-[info] %s\n"%(self.index,timeStr,strLine))
+            self.index += 1
+    
+    def writeConfig(self,config):
+        with open(self.path,'a+') as logf:
+            for item in config.items():
+                text = "[%d]-[parameters] %s--%s\n"%(self.index,item[0],str(item[1]))
+                logf.writelines(text)
+                self.index +=1
+    
+    def writeModel(self,modelText):
+        with open(self.path,'a+') as logf:
+            logf.writelines("[%d]-[model] %s\n"%(self.index,modelText))
+            self.index += 1
+    
+    def writeRawInfo(self, strLine):
+        with open(self.path,'a+') as logf:
+            timeStr = datetime.datetime.strftime(datetime.datetime.now(),self.timeStrFormat)
+            logf.writelines("[%d]-[info] %s\n"%(self.index,timeStr,strLine))
+            self.index += 1
+    
+    def writeTrainLog(self, epoch, step, logText):
+        with open(self.path,'a+') as logf:
+            timeStr = datetime.datetime.strftime(datetime.datetime.now(),self.timeStrFormat)
+            logf.writelines("[%d]-[%s]-[logInfo]-epoch[%d]-step[%d] %s\n"%(self.index,timeStr,epoch,step,logText))
+            self.index += 1