a9e42a7f1d0ea64be0371a6f7aff5039b5a48f4f,NodeClassifier/iterator.py,BatchIterator,gen_data,#BatchIterator#Any#Any#Any#Any#,60

Before Change

                model_outputs = session["model outputs"]

            rep_choice = np.random.choice([0,1])
            representation = list(model_outputs.values())[rep_choice]
            R[i] = representation

            packets = session["packets"]
            for j, packet in enumerate(packets):

After Change

            model_outputs = session["model outputs"]

            classification = model_outputs["classification"]
            classification = sorted(classification, key=lambda x: x[0])
            class_array = [p for c,p in classification]
            if mod_type == "label_swap":
                np.random.shuffle(class_array)
            R[i] = np.asarray(class_array)

Instances