724535172c099bb841a10914462d2a600d13ab24,tests/test_classify.py,TestClassify,test_em,#TestClassify#,56

Before Change

    def test_em(self):

        y, match_index = generate.simulate_features()

        train_df_matches = y.ix[match_index].sample(500)
        train_df_nonmatches = y.ix[y.index - match_index].sample(1500)
        train_df = train_df_matches.append(train_df_nonmatches)

        ecm = recordlinkage.ECMClassifier()

After Change

    def test_em(self):

        train_df_matches = self.y.ix[self.match_index].sample(500)
        train_df_nonmatches = self.y.ix[self.y.index.difference(self.match_index)].sample(1500)
        train_df = train_df_matches.append(train_df_nonmatches)

        ecm = recordlinkage.ECMClassifier()
        ecm.learn(train_df.round())
        ecm.predict(self.y.round())
        ecm.prob(self.y.round())

        self.assertTrue(ecm.p is not None)

Instances