Merge branch 'notebook_code' of https://github.com/corinne-hcr/e-miss…

…ion-eval-private-data into notebook_code
e-mission · Jul 19, 2021 · 1fba74a · 1fba74a
2 parents 6eb56b1 + 102b9bd
commit 1fba74a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tour_model_eval/build_save_model.py b/tour_model_eval/build_save_model.py
@@ -110,7 +110,7 @@ def main():
                     # compute unique label sets and their probabilities in one cluster
                     # 'p' refers to probability
                     unique_labels = user_label_df.groupby(user_label_df.columns.tolist()).size().reset_index(name='count')
-                    unique_labels['p'] = [unique_labels.iloc[i]['count'] / sum_trips for i in range(len(unique_labels))]
+                    unique_labels['p'] = unique_labels.count / sum_trips
                     labels_columns = user_label_df.columns.to_list()
                     for i in range(len(unique_labels)):
                         one_set_labels = {}
@@ -165,4 +165,3 @@ def main():
 
 
 
-