cognitive baysian train bug fix