def estimate(self, state, config): state.text = "World" def fit(self, state, config): print("Not training implementation") class AssemblerState(State): text = None class InputValidator(Stage): def operate(self, state, config): if state.text: raise ValueError("'text' is not None") if __name__ == "__main__": logging.basicConfig(level=logging.INFO) path = os.path.dirname(os.path.realpath(__file__)) app_config = Config() app_config.read_config_files([path + "/config.yaml"]) assembler = Assembler("Dump output example") assembler.set_stages([InputValidator(), WriteHello(path), WriteWorld(path)]) assembler.set_config(app_config) assembler.run(AssemblerState()) print("Hello output.txt contains '%s'" % open(path + hello_file_path, "r").read()) print("World output.txt contains '%s'" % open(path + world_file_path, "r").read())
class AssemblerState(State): training_message = None text = None class Formatter(Stage): def visualise(self, state, config): print("Visualiser result: %s" % state.training_message) if __name__ == "__main__": logging.basicConfig(level=logging.INFO) data = AssemblerState() assembler = Assembler("Training example") assembler.set_stages([HelloWorld(), Formatter()]) assembler.init_assembler() # Run assembler before training assembler.run(data, RunMode.TRAIN) print("Text before training is '%s'" % data.text) data.text = None # Clear text to prevent validation raising error # Run training mode assembler.run(data) # Run assembler after training assembler.run(data) print("Text after training is '%s'" % data.text)