Make sure LocationChoiceModel is in the urbansim.model namespace.

UDST · Apr 18, 2014 · 4193f12 · 4193f12
1 parent 3dd7877
commit 4193f12
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/urbansim/models/__init__.py b/urbansim/models/__init__.py
@@ -1 +1,2 @@
 from .hedonic import HedonicModel, HedonicModelGroup
+from .lcm import LocationChoiceModel
diff --git a/urbansim/models/lcm.py b/urbansim/models/lcm.py
@@ -61,7 +61,8 @@ def fit(self, choosers, alternatives, current_choice):
         """
         _, merged, chosen = interaction.mnl_interaction_dataset(
             choosers, alternatives, self.sample_size, current_choice)
-        model_design = dmatrix(self.model_expression, data=merged)
+        model_design = dmatrix(
+            self.model_expression, data=merged, return_type='dataframe')
         fit, results = mnl.mnl_estimate(
             model_design.as_matrix(), chosen, self.sample_size)
         self.fit_results = results