blob: 8c8fc5c4b117626609d555a413b36e194be2f180 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package ai.vespa.rankingexpression.importer.configmodelview;
import com.yahoo.tensor.Tensor;
import java.util.List;
import java.util.Map;
import java.util.Optional;
/**
* Config model view of an imported machine-learned model.
*
* @author bratseth
*/
public interface ImportedMlModel {
enum ModelType {
VESPA, XGBOOST, LIGHTGBM, TENSORFLOW, ONNX
}
String name();
String source();
ModelType modelType();
Optional<String> inputTypeSpec(String input);
@Deprecated(forRemoval = true)
Map<String, String> smallConstants();
@Deprecated(forRemoval = true)
Map<String, String> largeConstants();
Map<String, Tensor> smallConstantTensors();
Map<String, Tensor> largeConstantTensors();
Map<String, String> functions();
List<ImportedMlFunction> outputExpressions();
boolean isNative();
ImportedMlModel asNative();
}
|