Function get_data

metrics/plot.py:99–104

()

        # ADD DATA POINTS IN FEAT SPACE
        def get_data():
            data = tr_data[:plot_t + 1] if data_mode == 'tr' else eval_data[:plot_t + 1]
            d_x = torch.cat([t[1] for t in data], dim=0).cuda()  # [(),x,y] entry per task
            d_y = torch.cat([t[2] for t in data], dim=0)
            d_yunique = torch.unique(d_y).squeeze()
            return d_x, d_y.cpu().numpy(), d_yunique.cpu().numpy()

        def get_mem_data():
            d_x, d_y, d_yunique = [], [], []