def grenoble_estimation(main_data_dir): with open(main_data_dir+'grenoble.yml') as model_file,\ open(main_data_dir+'grenoble.csv') as data_file: model = choice_model.MultinomialLogit.from_yaml(model_file) model.load_data(data_file) interface = choice_model.PylogitInterface(model) interface.estimate() return interface
def simple_multinomial_pylogit_estimation(simple_multinomial_model_with_data): interface = choice_model.PylogitInterface( simple_multinomial_model_with_data) interface.estimate() return interface
def simple_multinomial_pylogit_interface(simple_multinomial_model_with_data): return choice_model.PylogitInterface(simple_multinomial_model_with_data)
def test_no_data(self, simple_multinomial_model): with pytest.raises(choice_model.interface.interface.NoDataLoaded): choice_model.PylogitInterface(simple_multinomial_model)
def test_simple_model(self, simple_model): with pytest.raises(TypeError): choice_model.PylogitInterface(simple_model)
def test_multinomial_logit(self, simple_multinomial_model_with_data): interface = choice_model.PylogitInterface( simple_multinomial_model_with_data) assert interface.model == simple_multinomial_model_with_data