diff options
author | Lester Solbakken <lesters@oath.com> | 2022-06-07 15:35:19 +0200 |
---|---|---|
committer | gjoranv <gv@verizonmedia.com> | 2022-06-08 11:45:31 +0200 |
commit | 8a01b3e2106739aaaf4ad918edbed04244360eb2 (patch) | |
tree | f7ea0d3cbb2417d619824d7e9b16d1e69c911e53 /config-model | |
parent | 438f313637fef3293661b9655636dc0a4b3d63d0 (diff) |
Remove TensorFlow feature converter
Diffstat (limited to 'config-model')
-rw-r--r-- | config-model/src/main/java/com/yahoo/schema/expressiontransforms/TensorFlowFeatureConverter.java | 27 |
1 files changed, 3 insertions, 24 deletions
diff --git a/config-model/src/main/java/com/yahoo/schema/expressiontransforms/TensorFlowFeatureConverter.java b/config-model/src/main/java/com/yahoo/schema/expressiontransforms/TensorFlowFeatureConverter.java index fd19c97036e..731f09011c4 100644 --- a/config-model/src/main/java/com/yahoo/schema/expressiontransforms/TensorFlowFeatureConverter.java +++ b/config-model/src/main/java/com/yahoo/schema/expressiontransforms/TensorFlowFeatureConverter.java @@ -23,9 +23,6 @@ import java.util.Map; */ public class TensorFlowFeatureConverter extends ExpressionTransformer<RankProfileTransformContext> { - /** A cache of imported models indexed by model path. This avoids importing the same model multiple times. */ - private final Map<Path, ConvertedModel> convertedTensorFlowModels = new HashMap<>(); - public TensorFlowFeatureConverter() {} @Override @@ -39,28 +36,10 @@ public class TensorFlowFeatureConverter extends ExpressionTransformer<RankProfil } private ExpressionNode transformFeature(ReferenceNode feature, RankProfileTransformContext context) { - if ( ! feature.getName().equals("tensorflow")) return feature; - - try { - FeatureArguments arguments = asFeatureArguments(feature.getArguments()); - ConvertedModel convertedModel = - convertedTensorFlowModels.computeIfAbsent(arguments.path(), - path -> ConvertedModel.fromSourceOrStore(path, false, context)); - return convertedModel.expression(arguments, context); - } - catch (IllegalArgumentException | UncheckedIOException e) { - throw new IllegalArgumentException("Could not use tensorflow model from " + feature, e); + if ( ! feature.getName().equals("tensorflow")) { + return feature; } - } - - private FeatureArguments asFeatureArguments(Arguments arguments) { - if (arguments.isEmpty()) - throw new IllegalArgumentException("A tensorflow node must take an argument pointing to " + - "the tensorflow model directory under [application]/models"); - if (arguments.expressions().size() > 3) - throw new IllegalArgumentException("A tensorflow feature can have at most 3 arguments"); - - return new FeatureArguments(arguments); + throw new IllegalArgumentException("Import of TensorFlow models is no longer supported"); } } |