Implement API to train classification models (#18475)

This commit is contained in:
Nicolas Mowen
2025-08-16 10:20:33 -05:00
committed by Blake Blackshear
parent 6dc36fcbb4
commit 2c7b71b16e
8 changed files with 219 additions and 19 deletions
+2 -2
View File
@@ -150,10 +150,10 @@ class EmbeddingMaintainer(threading.Thread):
)
)
for name, model_config in self.config.classification.custom.items():
for model_config in self.config.classification.custom.values():
self.realtime_processors.append(
CustomStateClassificationProcessor(
self.config, model_config, name, self.requestor, self.metrics
self.config, model_config, self.requestor, self.metrics
)
if model_config.state_config != None
else CustomObjectClassificationProcessor(