OrgLion-ML/golearn
1
0
Fork 0
mirror of https://github.com/sjwhitworth/golearn.git synced 2025-04-26 13:49:14 +08:00
Code Issues Releases Wiki Activity

catching nInstances == 0

Browse Source
This commit is contained in:
Ayush 2020-08-01 11:36:53 +05:30
parent 9d1ac82a40
commit 6a42fcd4ae
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
trees/cart_classifier.go
View File

@ -90,6 +90,9 @@ func computeEntropyAndModeLabel(y []int64, labels []int64) (float64, int64) {
} }
func calculateClassificationLoss(y []int64, labels []int64, criterion string) (float64, int64, error) { func calculateClassificationLoss(y []int64, labels []int64, criterion string) (float64, int64, error) {
if len(y) == 0 {
return 0, 0, errors.New("Need atleast 1 value to compute impurity")
}
if criterion == GINI { if criterion == GINI {
loss, modeLabel := computeGiniImpurityAndModeLabel(y, labels) loss, modeLabel := computeGiniImpurityAndModeLabel(y, labels)
return loss, modeLabel, nil return loss, modeLabel, nil