stpredictions.datasets package
Submodules
stpredictions.datasets.load_data module
Created on December 12, 2021
- stpredictions.datasets.load_data.load_bibtex(dir_path: str)
Load the bibtex dataset. __author__ = “Michael Gygli, ETH Zurich” from https://github.com/gyglim/dvn/blob/master/mlc_datasets/__init__.py
dir_path : string - containing location of bibtex.arff
- Xnp.array
Explanatory variables - N * 1836 array variables in one vector - e.g. ‘dependent’, ‘always’
- Ynp.array
Target variables - N * 159 array variables in one vector - e.g. ‘TAG_system’, ‘TAG_social_nets’
- X_txtlist
Explanatory variables - N * 1836 list variables in one vector - e.g. ‘dependent’, ‘always’
- Y_txtlist
Target variables - N * 159 list variables in one vector - e.g. ‘TAG_system’, ‘TAG_social_nets’
- stpredictions.datasets.load_data.load_corel5k(dir_path: str)
Load the bibtex dataset. __author__ = “Michael Gygli, ETH Zurich” from https://github.com/gyglim/dvn/blob/master/mlc_datasets/__init__.py
dir_path : string - containing location of bibtex.arff
- Xnp.array
Explanatory variables - N * 499 array variables in one vector - e.g. ‘dependent’, ‘always’
- Ynp.array
Target variables - N * 374 array variables in one vector - e.g. ‘TAG_system’, ‘TAG_social_nets’
- X_txtlist
Explanatory variables - N * 499 list variables in one vector - e.g. ‘dependent’, ‘always’
- Y_txtlist
Target variables - N * 374 list variables in one vector - e.g. ‘TAG_system’, ‘TAG_social_nets’