training_q.py 文件源码

def __init__(self, visuals):
        self.visuals = visuals
        if visuals:
            self.train_loss_list = []
            self.val_loss_list = []
            self.val_kappa_list = []
            self.val_accu_list = []
            self.epoch_list = []
            plt.ion()
            self.f, self.ax = plt.subplots(3, 1)
            red_line = mlines.Line2D([], [], color='red', markersize=15, label='Training loss')
            green_line = mlines.Line2D([], [], color='green', markersize=15, label='Validation loss')
            self.ax[0].legend(handles=[red_line, green_line], prop={'size': 8})
        store_training_logs.delete_file('run_script_logs.pkl')