Allow different lambdas across nests?

[Serena9070]

Hi Tianyu,

Thanks for the amazing package! I'm trying to allow different lambdas across nests by setting shared_lambda = False. However, it ran into the RuntimeError: Error(s) in loading state_dict for NestedLogitModel: Unexpected key(s) in state_dict: "lambdas". I wonder if there is a way to fix this. Thanks!

[Serena9070]

Hi Tianyu,

Thanks for updating the package to allow different lambdas across nests. Sorry, I have encountered another problem. When I tried setting "shared_lambda=False", I keep having the error: lambdas and lambda_weight should be the same. I wonder if there is a way to fix this or if you have any idea what's the potential cause. Thanks!

[TianyuDu]

Oops, that's a sanity check I implemented, and it seems to fail... Could you please attach a script that could replicate this issue? Ideally, with sample data, you can replace actual data with dummy values if the dataset is sensitive. Thank you!