Gini Impurity

Problem

Compute the Gini impurity of a list of class labels:

Gini = 1 - sum(p_i^2)

Empty input returns 0.0.

case 1

in: [[0,0,1,1]]

expected: 0.5

case 2

in: [[0,0,0,0]]

expected: 0

case 3

in: [[0,1,2,3]]

expected: 0.75

case 4

in: [[]]

expected: 0

Starter codePython

def gini_impurity(labels):
    # Your code here
    pass

4 test cases · run your code against them in the editor