您好，假如我用您的代码做单句话情感预测，是可以测试的吗，如何输出0,1或者概率？您的那个pb模型就可以吗？还是需要其他的，若是其他的pb模型，是需要更改输入的代码吗？