예제 #1
0
 def test_add(self):
   f = Features()
   f.append([1,2,3], 4)
   f.append(np.array([1,2,3]), np.array(4))
   assert len(f.measurements)==2
   f2 = Features()
   f2.extend([[1,2,3],[4,5,6]], [7,8])
   assert len(f2.measurements)==2
   f2.concatenate(f)
   assert len(f2.measurements)==4
예제 #2
0
    def _run(self):
        if self.args['output'] is None:
            self.args['output'] = str(self.args['primop']) + '-' + str(
                self.args['estimator']) + '.est'
        EstimatorClass = EstimatorRegistry.lookup(self.args['estimator'])
        est = EstimatorClass()

        f = Features()
        for feats_file in self.args['features']:
            f2 = Features()
            f2.read(feats_file)
            f.concatenate(f2)

        est.fit(f)
        self.data = est.get_params()
        self._save(self.data, self.args['output'])