/aiadv

Python library, from AiAdventures, providing utility functions for downloading datasets, etc.

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

aiadv

A small helper library to download the datasets that are used in our training programs at AiAdventures.

Install

pip install aiadv

How to use

There is only two things that you need to know

URLs class

This class saves the id and file name of all the available datasets

URLs.MOVIE_LENS_SAMPLE
{'id': '1k2y0qC0E3oHeGA5a427hRgfbW7hnQBgF', 'fname': 'movie_lens_sample.zip'}

untar_data function

This function will download the dataset for you and will return the path.

untar_data(URLs.YELP_REIVEWS)

That's it!