Skip to content

Commit 9ac9b57

Browse files
authored
Add the 'classification' inference config for pipelines
1 parent c59d4d9 commit 9ac9b57

File tree

3 files changed

+134
-21
lines changed

3 files changed

+134
-21
lines changed

output/schema/schema.json

Lines changed: 109 additions & 19 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

output/typescript/types.ts

Lines changed: 11 additions & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

specification/ingest/_types/Processors.ts

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -240,12 +240,25 @@ export class InferenceProcessor extends ProcessorBase {
240240
inference_config?: InferenceConfig
241241
}
242242

243+
/**
244+
* @variants container
245+
*/
243246
export class InferenceConfig {
244247
regression?: InferenceConfigRegression
248+
classification?: InferenceConfigClassification
245249
}
246250

247251
export class InferenceConfigRegression {
248-
results_field: string
252+
results_field?: Field
253+
num_top_feature_importance_values?: integer
254+
}
255+
256+
export class InferenceConfigClassification {
257+
num_top_classes?: integer
258+
num_top_feature_importance_values?: integer
259+
results_field?: Field
260+
top_classes_results_field?: Field
261+
prediction_field_type?: string
249262
}
250263

251264
export class JoinProcessor extends ProcessorBase {

0 commit comments

Comments
 (0)