def test_io(self): f = Features() f.append([1,2,3], 10) f.append([2,3,4], 11) f.append([3,4,5], 12) with in_temporary_directory(): f.write('test_file') f2 = Features() f2.read('test_file') assert np.all(f.array==f2.array)
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'])