search

MLBazaar / MLPrimitives

Primitives for machine learning and data science.
https://mlbazaar.github.io/MLPrimitives
MIT License
69 stars 38 forks source link

Error when loading the reviews dataset #230

Closed csala closed 4 years ago

csala commented 4 years ago

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'