(y)
| 410 | |
| 411 | |
| 412 | def calculate_entropy(y): |
| 413 | log2 = lambda x: math.log(x) / math.log(2) |
| 414 | unique_labels = np.unique(y) |
| 415 | entropy = 0 |
| 416 | for label in unique_labels: |
| 417 | count = len(y[y == label]) |
| 418 | p = count / len(y) |
| 419 | entropy += -p * log2(p) |
| 420 | return entropy |
| 421 | |
| 422 | |
| 423 | def calculate_gini(y): |
no outgoing calls
no test coverage detected