def test_end2endtraining(config):
"""Test anomaly detection training on public dataset."""
storage_adapter = SomStorageAdapter(config=config, feedback_strategy=None)
model_adapter = SomModelAdapter(storage_adapter=storage_adapter)
tc = SomTrainJob(node_map=2, model_adapter=model_adapter)
result, dist = tc.execute()
assert result == 0
def test_training_infer(config):
"""Test anomaly detection training on public dataset."""
model_adapter = SomModelAdapter(SomStorageAdapter(config=config, feedback_strategy=None))
tc_train = SomTrainJob(node_map=2, model_adapter=model_adapter, recreate_model=True)
result, dist = tc_train.execute()
assert result == 0
model_adapter = SomModelAdapter(SomStorageAdapter(config=config, feedback_strategy=None))
tc_infer = SomInferenceJob(model_adapter=model_adapter, sleep=False)
result = tc_infer.execute()
assert result == 0
def get_score(config, node_map, feedback):
"""Simple utility function for injecting custom mock function into Detector."""
feedback_strategy = FeedbackStrategy(config, func=feedback)
storage_adapter = SomStorageAdapter(config=config,
feedback_strategy=feedback_strategy)
model_adapter = SomModelAdapter(storage_adapter=storage_adapter)
tc = SomTrainJob(node_map=node_map, model_adapter=model_adapter)
success, dist = tc.execute()
freq_one = dist[-1]
return freq_one