在Tensorflow示例语音命令中计算精度和召回率

时间:2018-10-07 16:34:19

标签: python tensorflow precision confusion-matrix

我想在tensorflow的语音命令示例中计算精度和召回率:https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/speech_commands/train.py ,特别是在代码的这一部分。

      for i in xrange(0, set_size, FLAGS.batch_size):
test_fingerprints, test_ground_truth = audio_processor.get_data(
    FLAGS.batch_size, i, model_settings, 0.0, 0.0, 0, 'testing', sess)
test_accuracy, conf_matrix = sess.run(
    [evaluation_step, confusion_matrix],
    feed_dict={
        fingerprint_input: test_fingerprints,
        ground_truth_input: test_ground_truth,
        dropout_prob: 1.0
    })
batch_size = min(FLAGS.batch_size, set_size - i)
total_accuracy += (test_accuracy * batch_size) / set_size
if total_conf_matrix is None:
  total_conf_matrix = conf_matrix
else:
  total_conf_matrix += conf_matrix
tf.logging.info('Confusion Matrix:\n %s' % (total_conf_matrix))
tf.logging.info('Final test accuracy = %.1f%% (N=%d)' % (total_accuracy * 100,
                                                       set_size))

我有混淆矩阵,但是如何从中获得其他评估矩阵,例如精度和召回率?

谢谢

0 个答案:

没有答案