fix numpy bug in imputation example

fixes issue Imputation Training Example Err #2934 I filed.
Merge pull request #2935 from kevinpCroat/patch-1

FIX error in imputation example