models.py 文件源码

def __init__(self, echonet):
            assert isinstance(echonet, EchoNet)
            self.echonet = echonet
            self.loss = []
            self.train_score = []
            self.validation_score = []
            self.test_score = []
            self.start_time = None

            # self.f = plt.figure(figsize=(12, 12))
            self.cmap = sb.diverging_palette(220, 10, as_cmap=True)