Error when loading the reviews dataset
csala opened this issue · 0 comments
csala commented
The reviews dataset fails when loaded
In [1]: from mlprimitives.datasets import load_reviews
In [2]: reviews = load_reviews()
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-2-dcb45f42dd79> in <module>
----> 1 reviews = load_reviews()
~/Projects/MIT/MLPrimitives/mlprimitives/datasets.py in load_reviews()
328 y = X.pop('evaluation').values
329
--> 330 return Dataset(load_reviews.__doc__, X, y, r2_score)
331
332
TypeError: __init__() missing 2 required positional arguments: 'data_modality' and 'task_type'