search

DBrughmans / NICE

Apache License 2.0
22 stars 10 forks source link

An instance of NICE cannot be created #4

Closed alexvorobiev closed 1 year ago

alexvorobiev commented 2 years ago

I can't create an instance of NICE in Python-3.9.12:

In [3]: NICE_explainer = NICE(optimization='sparsity')
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
Input In [3], in <cell line: 1>()
----> 1 NICE_explainer = NICE(optimization='sparsity')

TypeError: __init__() missing 3 required positional arguments: 'predict_fn', 'X_train', and 'cat_feat'
Old-Shatterhand commented 1 year ago

Hi,

I fixed the example notebook in #5.

Best, Roman

DBrughmans commented 1 year ago

Starting from v0.2.0 we removed the .fit() method and included it in the .init() method. The notebook has now been updated to work with the new architecture by @Old-Shatterhand.