def get_pre_y(self):
# TODO ???
# pre_y = tf.reshape(tf.round(tf.sigmoid(self._output)), [-1])
pre_y = tf.arg_max(input=self._output, dimension=1)
return pre_y
微信
公众号
扫码关注公众号