envs.py 文件源码

def save_result(self, alpha, epochs, filename, save_path='./'):
        #self.result.to_csv(
        #    os.path.join(save_path,
        #                 filename + datetime.strftime(datetime.now(),
        #                                              '%Y%m%d-%H%M.csv'))
        #)

        ax = self.result.plot(title="Average adherence rate of patients",
                              legend=True,
                              yticks=[0.5, 0.6, 0.7, 0.8, 0.9])
        ax.set(xlabel = "alpha = {}, epochs = {}".format(alpha, epochs))
        fig = ax.get_figure()
        fig.savefig(
            os.path.join(save_path,
                         filename + datetime.strftime(datetime.now(),
                                                      'plot_%Y%m%d-%H%M.png'))
        )