19527: Fix Χ² calculation.

refs #19527

Arvados-DCO-1.1-Signed-off-by: Tom Clegg <tom@curii.com>

diff --git a/slicenumpy.go b/slicenumpy.go
index 6d8a531cbc866add507fefcb047cb2db7f72221b..e787693da573e3275193f5add2c3bbb1fb397dfd 100644 (file)
--- a/slicenumpy.go
+++ b/slicenumpy.go
@@ -266,11 +266,13 @@ func (cmd *sliceNumpy) run(prog string, args []string, stdin io.Reader, stdout,
                                }
                        }
                }
+               cmd.chi2Cases = nil
                cmd.trainingSetSize = 0
                for i := range cmd.cgnames {
                        if cmd.samples[i].isTraining {
                                cmd.trainingSet[i] = cmd.trainingSetSize
                                cmd.trainingSetSize++
+                               cmd.chi2Cases = append(cmd.chi2Cases, cmd.samples[i].isCase)
                        } else {
                                cmd.trainingSet[i] = -1
                        }