Python TeamSetFeatureExtractor Examples

Example #1

File: test_team_set_feature_extractor.py Project: RarefactionQ/DraftML

class TestTeamSetFeatureExtractor(unittest2.TestCase):
    def setUp(self):
        self.tsfe = TeamSetFeatureExtractor()
        self.tfe = TeamFeatureExtractor()

    def test_output(self):
        output = dict()

        teams = ["EHOME", "Alliance"]

        for name, value in zip(self.tsfe.extractFeatureNames(), self.tsfe.extract(teams)):
            output[name] = value

        self.assertEqual(len(self.tsfe.extractFeatureNames()), len(self.tsfe.extract(teams)))

        self.assertAlmostEqual(output["glicko2 rating_delta"], -65.08)
        self.assertAlmostEqual(output["glicko rating_delta"], 352.44)
        self.assertAlmostEqual(output["elo rating_delta"], 227.04)

Example #2

File: test_team_set_feature_extractor.py Project: RarefactionQ/DraftML

 def setUp(self):
     self.tsfe = TeamSetFeatureExtractor()
     self.tfe = TeamFeatureExtractor()