added cuda flag for GPU prediction

Signed-off-by: Zethson <[email protected]>

.cookietemple.yml

@@ -13,7 +13,7 @@ full_name: Lukas Heumos
 email: [email protected]
 project_name: lcep-package
 project_short_description: Python package for lcep.
-version: 1.1.0
+version: 1.2.0
 license: MIT
 command_line_interface: Click
 testing_library: pytest

CHANGELOG.rst

@@ -4,6 +4,19 @@ Changelog
 
 This project adheres to `Semantic Versioning <https://semver.org/>`_.
 
+1.2.0 (2020-08-28)
+------------------
+
+**Added**
+
+* Added flag --cuda and --no-cuda to enable and disable GPU prediction
+
+**Fixed**
+
+**Dependencies**
+
+**Deprecated**
+
 
 1.1.0 (2020-08-28)
 ------------------

cookietemple.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 1.1.0
+current_version = 1.2.0
 
 [bumpversion_files_whitelisted]
 setup_file = setup.py

docs/conf.py

@@ -53,9 +53,9 @@
 # the built documents.
 #
 # The short X.Y version.
-version = '1.1.0'
+version = '1.2.0'
 # The full version, including alpha/beta/rc tags.
-release = '1.1.0'
+release = '1.2.0'
 
 # The language for content autogenerated by Sphinx. Refer to documentation
 # for a list of supported languages.

lcep_package/__init__.py

@@ -2,4 +2,4 @@
 
 __author__ = """Lukas Heumos"""
 __email__ = '[email protected]'
-__version__ = '1.1.0'
+__version__ = '1.2.0'

lcep_package/cli.py

@@ -15,8 +15,9 @@
 @click.command()
 @click.option('-i', '--input', required=True, type=str, help='Path to data file to predict.')
 @click.option('-m', '--model', type=str, help='Path to an already trained XGBoost model. If not passed a default model will be loaded.')
+@click.option('-c/-nc', '--cuda/--no-cuda', type=bool, default=False, help='Whether to enable cuda or not')
 @click.option('-o', '--output', type=str, help='Path to write the output to')
-def main(input: str, model: str, output: str):
+def main(input: str, model: str, cuda: bool, output: str):
     """Console script for lcep-package."""
     print(r"""[bold blue]
     ██       ██████ ███████ ██████ 
@@ -32,6 +33,8 @@ def main(input: str, model: str, output: str):
         model = get_xgboost_model(f'{WD}/models/model_28.08.2020_v1.xgb')
     else:
         model = get_xgboost_model(model)
+    if cuda:
+        model.set_param({'predictor': 'gpu_predictor'})
     print('[bold blue] Parsing data')
     data_to_predict = read_data_to_predict(input)
     print('[bold blue] Performing predictions')

setup.py

@@ -72,6 +72,6 @@ def walker(base, *paths):
     test_suite='tests',
     tests_require=test_requirements,
     url='https://github.com/mlf-core/lcep-package',
-    version='1.1.0',
+    version='1.2.0',
     zip_safe=False,
 )