Commit 969237b9 authored by nsoler's avatar nsoler
Browse files

export_feature function for sign0

parent a77b6e64
Pipeline #2378 failed with stages
in 1 minute and 57 seconds
......@@ -511,7 +511,8 @@ class sign0(BaseSignature, DataSignature):
# After that check that your file is ok and move it to sign0.h5
self.__log.debug("Done")
def export_features(self,destination=self.model_path):
def export_features(self,destination=None):
features = self.features
destination = self.model_path if destination is None else destination
with h5py.File(os.path.join(destination, "features_sign0_"+self.dataset+".h5"), 'w') as hf_out:
hf_out.create_dataset("features", data=np.array(features, DataSignature.string_dtype()))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment