balajiln/mondrianforest

Key error

Opened this issue · 3 comments

Hey there

There seems to be a bug on example code mondrianforest_demo.py with PLOT=True :

Traceback (most recent call last):
  File "mondrianforest_demo.py", line 71, in <module>
    pred_mean = pred_forest_test['pred_mean']
KeyError: 'pred_mean'

pred_forest_test only has one key 'pred_prob'. The code tries to access both pred_mean and pred_var.

Manuel

Hi Manuel,

It seems like an issue related to incorrect optype. pred_mean and pred_sd are outputted for regression, pred_prob is outputted for classification. Can you send me the command you tried?

Balaji

Ah ok, thanks. I didn't test the code properly with plotting enabled, sorry about that. I'll fix it this weekend.

That is great! Let me know if you need any help.