Commit 45834156 authored by Pavel KORSHUNOV's avatar Pavel KORSHUNOV

fixing bugs

parent a149f371
Pipeline #15417 passed with stages
in 11 minutes and 48 seconds
...@@ -207,7 +207,11 @@ class TensorflowEval(Algorithm): ...@@ -207,7 +207,11 @@ class TensorflowEval(Algorithm):
for feat in feature: for feat in feature:
feat = numpy.cast['float32'](feat) feat = numpy.cast['float32'](feat)
self._check_feature(feat) self._check_feature(feat)
projections.extend(self.project_feature(feat)) projection = self.project_feature(feat)
if projection is not None:
if len(projections) == 0:
return None
return numpy.asarray(projections, dtype=numpy.float32) return numpy.asarray(projections, dtype=numpy.float32)
else: else:
feature = numpy.cast['float32'](feature) feature = numpy.cast['float32'](feature)
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment