Пример #1
0
 def accuracy_info(self):
     answerListB = self.testing_answer.tolist()
     answerList = [answer.index(1) for answer in answerListB]
     cm = ConfusionMatrix(answerList,
                          self.sess.run(self.predict_op,
                                        feed_dict={self.data_placeholder: self.testing_data,
                                                   self.answer_placeholder: self.testing_answer}))
     cmData = cm.to_array('a')
     acc = [cmData[0][0], cmData[1][1], cmData[2][2]]
     print cm
     print acc