Esempio n. 1
0
func entropyOverColumns(matrix [][]float64) float64 {
	returnValue, total := 0.0, 0.0
	var sumForColumn float64
	for j := range matrix[0] {
		sumForColumn = 0
		for i := range matrix {
			sumForColumn += matrix[i][j]
		}
		returnValue = returnValue - lnFunc(sumForColumn)
		total += sumForColumn
	}
	if utils.Eq(total, 0) {
		return 0
	}
	return (returnValue + lnFunc(total)) / (total * math.Log(2))
}
Esempio n. 2
0
func entropyConditionedOnRows(matrix [][]float64) float64 {
	returnValue, total := 0.0, 0.0
	var sumForRow float64
	for i := range matrix {
		sumForRow = 0
		for j := range matrix[0] {
			returnValue = returnValue + lnFunc(matrix[i][j])
			sumForRow += matrix[i][j]
		}
		returnValue = returnValue - lnFunc(sumForRow)
		total += sumForRow
	}
	if utils.Eq(total, 0) {
		return 0
	}
	return -returnValue / (total * math.Log(2))
}