From cf6192c81c33bec1b9972cc84504675ea6115d71 Mon Sep 17 00:00:00 2001 From: Ryan Schmukler Date: Tue, 28 Jun 2016 14:36:48 -0400 Subject: [PATCH] fix(id3): fix panic on SplitAttribute being nil --- trees/id3.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/trees/id3.go b/trees/id3.go index 26efc19..48a163a 100644 --- a/trees/id3.go +++ b/trees/id3.go @@ -113,7 +113,7 @@ func InferID3Tree(from base.FixedDataGrid, with RuleGenerator) *DecisionTreeNode // Generate the splitting rule splitRule := with.GenerateSplitRule(from) - if splitRule == nil { + if splitRule == nil || splitRule.SplitAttr == nil { // Can't determine, just return what we have return ret }