def setUp(self):
# create a dataset
self.dataset = Dataset(data={
"car_time":
array([
52.9, 4.10, 4.10, 56.20, 51.80, 0.20, 27.60, 89.90, 41.50,
95.00, 99.10, 18.50, 82.00, 8.60, 22.50, 51.40, 81.00, 51.00,
62.20, 95.10, 41.60
]),
"bus_time":
array([
4.4, 28.50, 86.90, 31.60, 20.20, 91.20, 79.70, 2.20, 24.50,
43.50, 8.40, 84.00, 38.00, 1.60, 74.10, 83.80, 19.20, 85.00,
90.10, 22.20, 91.50
]),
"choice":
array([
2, 2, 1, 2, 2, 1, 1, 2, 2, 2, 2, 1, 1, 2, 1, 1, 2, 1, 1, 2, 1
]),
"id":
arange(21) + 1
},
id_name="id",
dataset_name="person")
choices = Dataset(data={
"choice": array([1, 2]),
"names": array(["car", "bus"])
},
id_name="choice",
dataset_name="transport")
self.choicemodel = ChoiceModel(
choice_set=choices,
utilities="opus_core.linear_utilities",
probabilities="opus_core.mnl_probabilities",
choices="opus_core.random_choices")
self.specification = EquationSpecification(
coefficients=("beta1", "beta2", "beta2"),
variables=("constant", "biogeme.person_transport.time",
"biogeme.person_transport.time"),
equations=(1, 1, 2))
households.summary()
households.add_primary_attribute(data=[4,6,9,2,4,8,2,1,3,2], name="location")
households.get_attribute_names()
households.modify_attribute(name="location", data=[0,0], index=[0,1])
households.get_attribute("location")
households.get_data_element_by_id(5).location
#households.write_dataset(out_storage=storage, out_table_name="households_output")
households.get_dataset_name()
# Working with models
from opus_core.choice_model import ChoiceModel
choicemodel = ChoiceModel(choice_set=[1,2,3],
utilities = "opus_core.linear_utilities",
probabilities = "opus_core.mnl_probabilities",
choices = "opus_core.random_choices")
from numpy import array
from opus_core.equation_specification import EquationSpecification
specification = EquationSpecification(
coefficients = array([
"beta01", "beta12", "beta03", "beta13"
]),
variables = array([
"constant","household.persons", "constant", "household.persons"
]),
equations = array([
1, 2, 3, 3
])
)
