bug fix init nkw and nckw test and train