summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorJon Bratseth <bratseth@oath.com>2018-08-21 10:35:23 +0200
committerJon Bratseth <bratseth@oath.com>2018-08-21 10:35:23 +0200
commit61c3bef3dc00d485ca87cb2e2b145e1b20626bf7 (patch)
tree52f84d962cbc298535a68b21071e6fced8572e2e
parentb7ede5ee29906e0a1ea61aa42b56913b9d666736 (diff)
Change to inverted, safer test
-rw-r--r--config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/ConvertedModel.java16
-rw-r--r--config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/XgboostFeatureConverter.java2
2 files changed, 6 insertions, 12 deletions
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/ConvertedModel.java b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/ConvertedModel.java
index 5065c72a368..2bc0ccf6686 100644
--- a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/ConvertedModel.java
+++ b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/ConvertedModel.java
@@ -5,7 +5,6 @@ import com.yahoo.config.application.api.ApplicationFile;
import com.yahoo.config.application.api.ApplicationPackage;
import com.yahoo.config.model.application.provider.FilesApplicationPackage;
import com.yahoo.io.IOUtils;
-import com.yahoo.io.reader.NamedReader;
import com.yahoo.path.Path;
import com.yahoo.search.query.profile.QueryProfileRegistry;
import com.yahoo.searchdefinition.FeatureNames;
@@ -82,7 +81,7 @@ public class ConvertedModel {
this.modelPath = modelPath;
this.modelName = toModelName(modelPath);
ModelStore store = new ModelStore(context.rankProfile().getSearch().sourceApplication(), modelPath);
- if ( ! store.hasStoredModel()) // not converted yet - access from models/ directory
+ if ( store.hasSourceModel()) // not converted yet - access from models/ directory
expressions = importModel(store, context.rankProfile(), context.queryProfiles(), modelImporter, arguments);
else
expressions = transformFromStoredModel(store, context.rankProfile());
@@ -93,7 +92,7 @@ public class ConvertedModel {
QueryProfileRegistry queryProfiles,
ModelImporter modelImporter,
FeatureArguments arguments) {
- ImportedModel model = modelImporter.importModel(store.modelFiles.modelName(), store.modelDir());
+ ImportedModel model = modelImporter.importModel(store.modelFiles.modelName(), store.sourceModelDir());
return transformFromImportedModel(model, store, profile, queryProfiles, arguments);
}
@@ -501,19 +500,14 @@ public class ConvertedModel {
this.modelFiles = new ModelFiles(modelPath);
}
- public boolean hasStoredModel() {
- try {
- return application.getFileReference(modelFiles.storedModelPath()).exists();
- }
- catch (UnsupportedOperationException e) {
- return false;
- }
+ public boolean hasSourceModel() {
+ return sourceModelDir().exists();
}
/**
* Returns the directory which contains the source model to use for these arguments
*/
- public File modelDir() {
+ public File sourceModelDir() {
return application.getFileReference(ApplicationPackage.MODELS_DIR.append(modelFiles.modelPath()));
}
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/XgboostFeatureConverter.java b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/XgboostFeatureConverter.java
index 62f43e15849..11f64e68519 100644
--- a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/XgboostFeatureConverter.java
+++ b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/XgboostFeatureConverter.java
@@ -39,7 +39,7 @@ public class XgboostFeatureConverter extends ExpressionTransformer<RankProfileTr
ConvertedModel.FeatureArguments arguments = asFeatureArguments(feature.getArguments());
ConvertedModel.ModelStore store = new ConvertedModel.ModelStore(context.rankProfile().getSearch().sourceApplication(),
arguments.modelPath());
- RankingExpression expression = xgboostImporter.parseModel(store.modelDir().toString());
+ RankingExpression expression = xgboostImporter.parseModel(store.sourceModelDir().toString());
return expression.getRoot();
} catch (IllegalArgumentException | UncheckedIOException e) {
throw new IllegalArgumentException("Could not use XGBoost model from " + feature, e);