def test_seq2seq_hits1(self):
        import projects.convai2.baselines.seq2seq.eval_hits as eval_hits

        report = eval_hits.main(args=[])
        self.assertAlmostEqual(report['hits@1'], 0.1247, places=4)
Example #2
0
    def test_kvmemnn_hits1(self):
        import projects.convai2.baselines.kvmemnn.eval_hits as eval_hits

        with testing_utils.capture_output() as stdout:
            report = eval_hits.main()
        self.assertEqual(report['hits@1'], 0.5510, str(stdout))
    def test_seq2seq_hits1(self):
        import projects.convai2.baselines.seq2seq.eval_hits as eval_hits

        with testing_utils.capture_output() as stdout:
            report = eval_hits.main()
        self.assertEqual(report['hits@1'], .1250, str(stdout))