Ejemplos de calculate_entropy en Python

Ejemplo n.º 1

Archivo: decision_tree.py Proyecto: ravindra-kumar25/machine-learning-basics

    def _calculate_information_gain(self, y, y1, y2):
        # Calculate information gain
        p = len(y1) / len(y)
        entropy = calculate_entropy(y)
        info_gain = entropy - p * calculate_entropy(y1) - (1 - p) * calculate_entropy(y2)

        return info_gain

Ejemplo n.º 2

Archivo: decision_tree.py Proyecto: PSEUDOBUBLAR/ML-From-Scratch

    def _calculate_information_gain(self, y, y1, y2):
        # Calculate information gain
        p = len(y1) / len(y)
        entropy = calculate_entropy(y)
        info_gain = entropy - p * \
            calculate_entropy(y1) - (1 - p) * \
            calculate_entropy(y2)

        return info_gain

Ejemplo n.º 3

Archivo: Random_Forest_classifier.py Proyecto: comcidis/CPE-WSCAD2020-Extend

 def _calculate_information_gain(self, y, y1, y2):
     with pyRAPL.Measurement('Information_gain', output=csv_output):
         # Calculate information gain
         p = len(y1) / len(y)
         entropy = calculate_entropy(y)
         info_gain = entropy - p * \
            calculate_entropy(y1) - (1 - p) * \
            calculate_entropy(y2)
         return info_gain
     csv_output.save()