3436982b68daae3a0195226b72a3d7cd487793d7,basic/evaluator.py,F1Evaluator,get_evaluation,#F1Evaluator#Any#Any#,204

Before Change

    def get_evaluation(self, sess, batch):
        idxs, data_set = batch
        assert isinstance(data_set, DataSet)
        feed_dict = self.model.get_feed_dict(data_set, False)
        global_step, yp, yp2, loss = sess.run([self.model.global_step, self.model.yp, self.model.yp2, self.model.loss], feed_dict=feed_dict)
        y = data_set.data["y"]
        if self.config.squash:
            new_y = []

After Change

class F1Evaluator(LabeledEvaluator):
    def get_evaluation(self, sess, batch):
        idxs, data_set = self._split_batch(batch)
        assert isinstance(data_set, DataSet)
        feed_dict = self._get_feed_dict(batch)
        global_step, yp, yp2, loss = sess.run([self.model.global_step, self.model.yp, self.model.yp2, self.model.loss], feed_dict=feed_dict)
        y = data_set.data["y"]
        if self.config.squash:
            new_y = []

Instances