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

Before Change

            session = sessions[key][session_id]
            if mod_type == "rep_swap":
                new_key = np.random.choice(key_subset)
                new_id = np.random.choice(len(sessions[new_key]))
                new_session = sessions[new_key][new_id]
                model_outputs = new_session["model outputs"]
            else:
                model_outputs = session["model outputs"]

After Change

        hex_str = "0123456789abcdef"
        mod_types = ["label_swap", "packet_swap", "duplicate"]
        if self.perturb_types != "all":
            mod_types = self.perturb_types
        if length <= 1:
            mod_types = ["label_swap"]

        all_keys = list(self.sessions_by_length[length].keys())

Instances