Ejemplo n.º 1
0
    def add_itemset(self, itemset_pass):
        item = None
        antecedent_array = [0 for x in range(self.n_variables)]
        for i in range(0, self.n_variables):
            antecedent_array[i] = -1
        for i in range(0, itemset_pass.size()):
            item = itemset_pass.get(i)
            antecedent_array[item.get_variable()] = item.get_label()

        rule = Rule(self.data_base)
        rule.assign_antecedente(antecedent_array)
        rule.set_consequent(itemset_pass.get_class())
        rule.set_confidence(itemset_pass.get_support_class() / itemset_pass.get_support())
        rule.set_support(itemset_pass.get_support_class())
        self.rule_base_array.append(rule)