inference.asciidoc 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495
  1. [role="xpack"]
  2. [[inference-processor]]
  3. === {infer-cap} processor
  4. ++++
  5. <titleabbrev>{infer-cap}</titleabbrev>
  6. ++++
  7. Uses a pre-trained {dfanalytics} model or a model deployed for natural
  8. language processing tasks to infer against the data that is being
  9. ingested in the pipeline.
  10. [[inference-options]]
  11. .{infer-cap} Options
  12. [options="header"]
  13. |======
  14. | Name | Required | Default | Description
  15. | `model_id` . | yes | - | (String) The ID or alias for the trained model, or the ID of the deployment.
  16. | `target_field` | no | `ml.inference.<processor_tag>` | (String) Field added to incoming documents to contain results objects.
  17. | `field_map` | no | If defined the model's default field map | (Object) Maps the document field names to the known field names of the model. This mapping takes precedence over any default mappings provided in the model configuration.
  18. | `inference_config` | no | The default settings defined in the model | (Object) Contains the inference type and its options.
  19. include::common-options.asciidoc[]
  20. |======
  21. [source,js]
  22. --------------------------------------------------
  23. {
  24. "inference": {
  25. "model_id": "model_deployment_for_inference",
  26. "target_field": "FlightDelayMin_prediction_infer",
  27. "field_map": {
  28. "your_field": "my_field"
  29. },
  30. "inference_config": { "regression": {} }
  31. }
  32. }
  33. --------------------------------------------------
  34. // NOTCONSOLE
  35. [discrete]
  36. [[inference-processor-classification-opt]]
  37. ==== {classification-cap} configuration options
  38. Classification configuration for inference.
  39. `num_top_classes`::
  40. (Optional, integer)
  41. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-classification-num-top-classes]
  42. `num_top_feature_importance_values`::
  43. (Optional, integer)
  44. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-classification-num-top-feature-importance-values]
  45. `results_field`::
  46. (Optional, string)
  47. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-results-field-processor]
  48. `top_classes_results_field`::
  49. (Optional, string)
  50. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-classification-top-classes-results-field]
  51. `prediction_field_type`::
  52. (Optional, string)
  53. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-classification-prediction-field-type]
  54. [discrete]
  55. [[inference-processor-fill-mask-opt]]
  56. ==== Fill mask configuration options
  57. `num_top_classes`::
  58. (Optional, integer)
  59. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-classification-num-top-classes]
  60. `results_field`::
  61. (Optional, string)
  62. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-results-field-processor]
  63. `tokenization`::
  64. (Optional, object)
  65. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization]
  66. +
  67. .Properties of tokenization
  68. [%collapsible%open]
  69. =====
  70. `bert`::::
  71. (Optional, object)
  72. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-bert]
  73. +
  74. .Properties of bert
  75. [%collapsible%open]
  76. =======
  77. `truncate`::::
  78. (Optional, string)
  79. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  80. =======
  81. `roberta`::::
  82. (Optional, object)
  83. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-roberta]
  84. +
  85. .Properties of roberta
  86. [%collapsible%open]
  87. =======
  88. `truncate`::::
  89. (Optional, string)
  90. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  91. =======
  92. `mpnet`::::
  93. (Optional, object)
  94. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-mpnet]
  95. +
  96. .Properties of mpnet
  97. [%collapsible%open]
  98. =======
  99. `truncate`::::
  100. (Optional, string)
  101. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  102. =======
  103. =====
  104. [discrete]
  105. [[inference-processor-ner-opt]]
  106. ==== NER configuration options
  107. `results_field`::
  108. (Optional, string)
  109. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-results-field-processor]
  110. `tokenization`::
  111. (Optional, object)
  112. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization]
  113. +
  114. .Properties of tokenization
  115. [%collapsible%open]
  116. =====
  117. `bert`::::
  118. (Optional, object)
  119. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-bert]
  120. +
  121. .Properties of bert
  122. [%collapsible%open]
  123. =======
  124. `truncate`::::
  125. (Optional, string)
  126. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  127. =======
  128. `roberta`::::
  129. (Optional, object)
  130. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-roberta]
  131. +
  132. .Properties of roberta
  133. [%collapsible%open]
  134. =======
  135. `truncate`::::
  136. (Optional, string)
  137. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  138. =======
  139. `mpnet`::::
  140. (Optional, object)
  141. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-mpnet]
  142. +
  143. .Properties of mpnet
  144. [%collapsible%open]
  145. =======
  146. `truncate`::::
  147. (Optional, string)
  148. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  149. =======
  150. =====
  151. [discrete]
  152. [[inference-processor-regression-opt]]
  153. ==== {regression-cap} configuration options
  154. Regression configuration for inference.
  155. `results_field`::
  156. (Optional, string)
  157. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-results-field-processor]
  158. `num_top_feature_importance_values`::
  159. (Optional, integer)
  160. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-regression-num-top-feature-importance-values]
  161. [discrete]
  162. [[inference-processor-text-classification-opt]]
  163. ==== Text classification configuration options
  164. `classification_labels`::
  165. (Optional, string) An array of classification labels.
  166. `num_top_classes`::
  167. (Optional, integer)
  168. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-classification-num-top-classes]
  169. `results_field`::
  170. (Optional, string)
  171. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-results-field-processor]
  172. `tokenization`::
  173. (Optional, object)
  174. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization]
  175. +
  176. .Properties of tokenization
  177. [%collapsible%open]
  178. =====
  179. `bert`::::
  180. (Optional, object)
  181. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-bert]
  182. +
  183. .Properties of bert
  184. [%collapsible%open]
  185. =======
  186. `span`::::
  187. (Optional, integer)
  188. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-span]
  189. `truncate`::::
  190. (Optional, string)
  191. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  192. =======
  193. `roberta`::::
  194. (Optional, object)
  195. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-roberta]
  196. +
  197. .Properties of roberta
  198. [%collapsible%open]
  199. =======
  200. `span`::::
  201. (Optional, integer)
  202. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-span]
  203. `truncate`::::
  204. (Optional, string)
  205. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  206. =======
  207. `mpnet`::::
  208. (Optional, object)
  209. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-mpnet]
  210. +
  211. .Properties of mpnet
  212. [%collapsible%open]
  213. =======
  214. `truncate`::::
  215. (Optional, string)
  216. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  217. =======
  218. =====
  219. [discrete]
  220. [[inference-processor-text-embedding-opt]]
  221. ==== Text embedding configuration options
  222. `results_field`::
  223. (Optional, string)
  224. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-results-field-processor]
  225. `tokenization`::
  226. (Optional, object)
  227. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization]
  228. +
  229. .Properties of tokenization
  230. [%collapsible%open]
  231. =====
  232. `bert`::::
  233. (Optional, object)
  234. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-bert]
  235. +
  236. .Properties of bert
  237. [%collapsible%open]
  238. =======
  239. `truncate`::::
  240. (Optional, string)
  241. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  242. =======
  243. `roberta`::::
  244. (Optional, object)
  245. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-roberta]
  246. +
  247. .Properties of roberta
  248. [%collapsible%open]
  249. =======
  250. `truncate`::::
  251. (Optional, string)
  252. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  253. =======
  254. `mpnet`::::
  255. (Optional, object)
  256. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-mpnet]
  257. +
  258. .Properties of mpnet
  259. [%collapsible%open]
  260. =======
  261. `truncate`::::
  262. (Optional, string)
  263. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  264. =======
  265. =====
  266. [discrete]
  267. [[inference-processor-zero-shot-opt]]
  268. ==== Zero shot classification configuration options
  269. `labels`::
  270. (Optional, array)
  271. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-zero-shot-classification-labels]
  272. `multi_label`::
  273. (Optional, boolean)
  274. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-zero-shot-classification-multi-label]
  275. `results_field`::
  276. (Optional, string)
  277. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-results-field-processor]
  278. `tokenization`::
  279. (Optional, object)
  280. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization]
  281. +
  282. .Properties of tokenization
  283. [%collapsible%open]
  284. =====
  285. `bert`::::
  286. (Optional, object)
  287. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-bert]
  288. +
  289. .Properties of bert
  290. [%collapsible%open]
  291. =======
  292. `truncate`::::
  293. (Optional, string)
  294. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  295. =======
  296. `roberta`::::
  297. (Optional, object)
  298. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-roberta]
  299. +
  300. .Properties of roberta
  301. [%collapsible%open]
  302. =======
  303. `truncate`::::
  304. (Optional, string)
  305. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  306. =======
  307. `mpnet`::::
  308. (Optional, object)
  309. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-mpnet]
  310. +
  311. .Properties of mpnet
  312. [%collapsible%open]
  313. =======
  314. `truncate`::::
  315. (Optional, string)
  316. include::{es-repo-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization-truncate]
  317. =======
  318. =====
  319. [discrete]
  320. [[inference-processor-config-example]]
  321. ==== {infer-cap} processor examples
  322. [source,js]
  323. --------------------------------------------------
  324. "inference":{
  325. "model_id": "my_model_id",
  326. "field_map": {
  327. "original_fieldname": "expected_fieldname"
  328. },
  329. "inference_config": {
  330. "regression": {
  331. "results_field": "my_regression"
  332. }
  333. }
  334. }
  335. --------------------------------------------------
  336. // NOTCONSOLE
  337. This configuration specifies a `regression` inference and the results are
  338. written to the `my_regression` field contained in the `target_field` results
  339. object. The `field_map` configuration maps the field `original_fieldname` from
  340. the source document to the field expected by the model.
  341. [source,js]
  342. --------------------------------------------------
  343. "inference":{
  344. "model_id":"my_model_id"
  345. "inference_config": {
  346. "classification": {
  347. "num_top_classes": 2,
  348. "results_field": "prediction",
  349. "top_classes_results_field": "probabilities"
  350. }
  351. }
  352. }
  353. --------------------------------------------------
  354. // NOTCONSOLE
  355. This configuration specifies a `classification` inference. The number of
  356. categories for which the predicted probabilities are reported is 2
  357. (`num_top_classes`). The result is written to the `prediction` field and the top
  358. classes to the `probabilities` field. Both fields are contained in the
  359. `target_field` results object.
  360. For an example that uses {nlp} trained models, refer to
  361. {ml-docs}/ml-nlp-inference.html[Add NLP inference to ingest pipelines].
  362. [discrete]
  363. [[inference-processor-feature-importance]]
  364. ==== {feat-imp-cap} object mapping
  365. To get the full benefit of aggregating and searching for
  366. {ml-docs}/ml-feature-importance.html[{feat-imp}], update your index mapping of
  367. the {feat-imp} result field as you can see below:
  368. [source,js]
  369. --------------------------------------------------
  370. "ml.inference.feature_importance": {
  371. "type": "nested",
  372. "dynamic": true,
  373. "properties": {
  374. "feature_name": {
  375. "type": "keyword"
  376. },
  377. "importance": {
  378. "type": "double"
  379. }
  380. }
  381. }
  382. --------------------------------------------------
  383. // NOTCONSOLE
  384. The mapping field name for {feat-imp} (in the example above, it is
  385. `ml.inference.feature_importance`) is compounded as follows:
  386. `<ml.inference.target_field>`.`<inference.tag>`.`feature_importance`
  387. * `<ml.inference.target_field>`: defaults to `ml.inference`.
  388. * `<inference.tag>`: if is not provided in the processor definition, then it is
  389. not part of the field path.
  390. For example, if you provide a tag `foo` in the definition as you can see below:
  391. [source,js]
  392. --------------------------------------------------
  393. {
  394. "tag": "foo",
  395. ...
  396. }
  397. --------------------------------------------------
  398. // NOTCONSOLE
  399. Then, the {feat-imp} value is written to the
  400. `ml.inference.foo.feature_importance` field.
  401. You can also specify the target field as follows:
  402. [source,js]
  403. --------------------------------------------------
  404. {
  405. "tag": "foo",
  406. "target_field": "my_field"
  407. }
  408. --------------------------------------------------
  409. // NOTCONSOLE
  410. In this case, {feat-imp} is exposed in the
  411. `my_field.foo.feature_importance` field.