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