MCPcopy Create free account
hub / github.com/catboost/catboost / higgs

Function higgs

catboost/python-package/catboost/datasets.py:320–336  ·  view source on GitHub ↗

Download "higgs" [1] data set. Will return two pandas.DataFrame-s, first with train part and second with test part of the dataset. Object class will be located in the first column of dataset. [1]: https://archive.ics.uci.edu/ml/datasets/HIGGS

()

Source from the content-addressed store, hash-verified

318
319
320def higgs():
321 """
322 Download "higgs" [1] data set.
323
324 Will return two pandas.DataFrame-s, first with train part and second with
325 test part of the dataset. Object class will be located in the first
326 column of dataset.
327
328 [1]: https://archive.ics.uci.edu/ml/datasets/HIGGS
329 """
330 url = 'https://storage.mds.yandex.net/get-devtools-opensource/250854/higgs.tar.gz'
331 md5 = 'ad59ba8328a9afa3837d7bf1a0e10e7b'
332 dataset_name, train_file, test_file = 'higgs', 'train.tsv', 'test.tsv'
333 train_path, test_path = _download_dataset(url, md5, dataset_name, train_file, test_file, cache=True)
334 return (
335 _load_numeric_only_dataset(train_path, 10500000, 29, sep='\t'),
336 _load_numeric_only_dataset(test_path, 500000, 29, sep='\t'))

Callers

nothing calls this directly

Calls 2

_download_datasetFunction · 0.85

Tested by

no test coverage detected