tensorflow+bert文本二分类0-1,model.predict之后获得的结果
#predict using the model
#references
a=model.predict(ds_test_encoded,batch_size=1)
a如下
(array([[ 6.1176378e-01, -1.1029058e+00],
[ 1.0485318e+00, -1.3658700e+00],
请问如何转化为0-1结果呢
tensorflow+bert文本二分类0-1,model.predict之后获得的结果
#predict using the model
#references
a=model.predict(ds_test_encoded,batch_size=1)
a如下
(array([[ 6.1176378e-01, -1.1029058e+00],
[ 1.0485318e+00, -1.3658700e+00],
请问如何转化为0-1结果呢