|
|
|
@ -2,25 +2,20 @@
|
|
|
|
|
# Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved.
|
|
|
|
|
# Released under the modified BSD license. See COPYING.md for more details.
|
|
|
|
|
|
|
|
|
|
from abc import ABC, abstractmethod
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Component(ABC):
|
|
|
|
|
class Component:
|
|
|
|
|
"""
|
|
|
|
|
A Component is an object which adds functionality to a LearningSolver.
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
@abstractmethod
|
|
|
|
|
def before_solve(self, solver, instance, model):
|
|
|
|
|
pass
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
@abstractmethod
|
|
|
|
|
def after_solve(self, solver, instance, model, results):
|
|
|
|
|
pass
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
@abstractmethod
|
|
|
|
|
def fit(self, training_instances):
|
|
|
|
|
pass
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
def iteration_cb(self, solver, instance, model):
|
|
|
|
|
return False
|
|
|
|
|