Error in example code / NAMdataset

[pkopper]

The example code is not complete and missing the pandas import call. (import pandas as pd)
Additionally, I suggest to actually import sklearn.datasets explicitly.
If done so, I still cannot source your example code due to an error thrown in NAMdataset.

precisely:

dataset = NAMDataset(config,
                      data_path=dataset,
                      features_columns=dataset.columns[:-1],
                      targets_column=dataset.columns[-1])

results in

Traceback (most recent call last):

  File ~/.../site-packages/nam/data/base.py:76 in __init__
    self.features, self.features_names = transform_data(self.raw_X)

  File ~/.../site-packages/nam/data/utils.py:86 in transform_data
    cat_ohe_step = ('ohe', OneHotEncoder(sparse=False, handle_unknown='ignore'))

TypeError: OneHotEncoder.__init__() got an unexpected keyword argument 'sparse'