Log dimensions.

Arvados-DCO-1.1-Signed-off-by: Tom Clegg <tom@tomclegg.ca>

diff --git a/pca.go b/pca.go
index d32751ea6d013a49386d3a6477f3bccc5ec664ae..b23838d771060d9f0eb4168ba8380b2f1ff16f2a 100644 (file)
--- a/pca.go
+++ b/pca.go
@@ -152,9 +152,10 @@ func (cmd *goPCA) RunCommand(prog string, args []string, stdin io.Reader, stdout
        log.Print("converting cgs to array")
        data, rows, cols := cgs2array(cgs)
        if *onehot {
+               log.Printf("recode one-hot: %d rows, %d cols", rows, cols)
                data, cols = recodeOnehot(data, cols)
        }
-       log.Print("running fit+transform")
+       log.Printf("running fit+transform: %d rows, %d cols", rows, cols)
        pca, err := nlp.NewPCA(*components).FitTransform(array2matrix(rows, cols, data).T())
        if err != nil {
                return 1
@@ -162,8 +163,8 @@ func (cmd *goPCA) RunCommand(prog string, args []string, stdin io.Reader, stdout
 
        log.Print("transposing result")
        pca = pca.T()
-       log.Print("copying result to numpy output array")
        rows, cols = pca.Dims()
+       log.Printf("copying result to numpy output array: %d rows, %d cols", rows, cols)
        out := make([]float64, rows*cols)
        for i := 0; i < rows; i++ {
                for j := 0; j < cols; j++ {
@@ -187,7 +188,7 @@ func (cmd *goPCA) RunCommand(prog string, args []string, stdin io.Reader, stdout
                return 1
        }
        npw.Shape = []int{rows, cols}
-       log.Print("writing numpy")
+       log.Printf("writing numpy: %d rows, %d cols", rows, cols)
        npw.WriteFloat64(out)
        err = bufw.Flush()
        if err != nil {