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)
