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
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'])