def test_load_and_persist_without_train(component_builder): _config = utilities.base_test_conf("all_components") trainer = Trainer(_config, component_builder) persistor = create_persistor(_config) persisted_path = trainer.persist(_config['path'], persistor, model_name=_config['name']) loaded = utilities.load_interpreter_for_model(_config, persisted_path, component_builder) assert loaded.pipeline assert loaded.parse("hello", time=None) is not None
def test_load_and_persist_without_train(component_builder): _config = utilities.base_test_conf("all_components") trainer = Trainer(_config, component_builder) persistor = create_persistor(_config) persisted_path = trainer.persist(_config['path'], persistor, project_name=_config['name']) loaded = utilities.load_interpreter_for_model(_config, persisted_path, component_builder) assert loaded.pipeline assert loaded.parse("hello") is not None assert loaded.parse("Hello today is Monday, again!") is not None
def test_load_and_persist_without_train(component_builder, tmpdir): _config = utilities.base_test_conf("all_components") trainer = Trainer(_config, component_builder) persistor = create_persistor(_config) persisted_path = trainer.persist(tmpdir.strpath, persistor, project_name="my_project") loaded = Interpreter.load(persisted_path, component_builder) assert loaded.pipeline assert loaded.parse("hello") is not None assert loaded.parse("Hello today is Monday, again!") is not None
def test_train_with_empty_data(component_builder): _config = utilities.base_test_conf("all_components") trainer = Trainer(_config, component_builder) trainer.train(TrainingData()) persistor = create_persistor(_config) persisted_path = trainer.persist(_config['path'], persistor, model_name=_config['name']) loaded = utilities.load_interpreter_for_model(_config, persisted_path, component_builder) assert loaded.pipeline assert loaded.parse("hello") is not None assert loaded.parse("Hello today is Monday, again!") is not None
def train(cfg_name, model_name): from rasa_nlu.train import create_persistor from rasa_nlu.converters import load_data config = RasaNLUConfig(cfg_name) trainer = Trainer(config) training_data = load_data(config['data']) trainer.train(training_data) persistor = create_persistor(config) trainer.persist("test_models", persistor, model_name=model_name)
def train(cfg_name, project_name): from rasa_nlu.train import create_persistor from rasa_nlu import training_data config = RasaNLUConfig(cfg_name) trainer = Trainer(config, component_builder) training_data = training_data.load_data(config['data']) trainer.train(training_data) persistor = create_persistor(config) trainer.persist("test_projects", persistor, project_name)
def test_train_with_empty_data(language, pipeline, component_builder, tmpdir): _config = RasaNLUModelConfig({"pipeline": pipeline, "language": language}) trainer = Trainer(_config, component_builder) trainer.train(TrainingData()) persistor = create_persistor(_config) persisted_path = trainer.persist(tmpdir.strpath, persistor, project_name="my_project") loaded = Interpreter.load(persisted_path, component_builder) assert loaded.pipeline assert loaded.parse("hello") is not None assert loaded.parse("Hello today is Monday, again!") is not None
def test_train_with_empty_data(component_builder): _config = utilities.base_test_conf("all_components") trainer = Trainer(_config, component_builder) trainer.train(TrainingData()) persistor = create_persistor(_config) persisted_path = trainer.persist(_config['path'], persistor, model_name=_config['name']) loaded = utilities.load_interpreter_for_model(_config, persisted_path, component_builder) assert loaded.pipeline assert loaded.parse("hello", time=None) is not None
def train(cfg_name, model_name): from rasa_nlu.train import create_trainer from rasa_nlu.train import create_persistor from rasa_nlu.training_data import TrainingData config = RasaNLUConfig(cfg_name) trainer = create_trainer(config) persistor = create_persistor(config) training_data = TrainingData(config['data'], config['backend'], nlp=trainer.nlp) trainer.train(training_data) trainer.persist(os.path.join("test_models", model_name), persistor, create_unique_subfolder=False)
def train(cfg_name, model_name): from rasa_nlu.train import create_persistor from rasa_nlu.converters import load_data config = RasaNLUConfig(cfg_name) trainer = Trainer(config) training_data = load_data(config['data']) trainer.train(training_data) persistor = create_persistor(config) trainer.persist(os.path.join("test_models", model_name), persistor, create_unique_subfolder=False)