These doesn't match your parameters. Where is DATA? It seems like you mix up the docstring with load_data.