# (C) Copyright IBM Corp. 2020.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

PYSPARK = 'pyspark'
SCIKIT = 'scikit-learn'
PANDAS = 'pandas'
XGBOOST= 'xgboost'
MLPIPELINE = 'mlpipelinepy'
IBMSPARKPIPELINE = 'ibmsparkpipeline'
TENSORFLOW = 'tensorflow'
HYBRID = 'hybrid'
SPSS_MODELER = 'spss-modeler'
PMML = 'pmml'
CAFFE = 'caffe'
CAFFE2 = 'caffe2'
PYTORCH = 'pytorch'
BLUECONNECT = 'blueconnect'
TORCH = 'torch'
MXNET = 'mxnet'
THEANO = 'theano'
DARKNET = 'darknet'

DISPLAY_PYSPARK = 'PySpark'
DISPLAY_SCIKIT = 'scikit-learn'
DISPLAY_PANDAS = 'Pandas'
DISPLAY_XGBOOST = 'XGBoost'
DISPLAY_MLPIPELINE = 'MLPipeline'
DISPLAY_IBMSPARKPIPELINE = 'IBMSparkPipeline'
DISPLAY_TENSORFLOW = 'Tensorflow'
DISPLAY_PMML = 'PMML'
DISPLAY_CAFFE = 'Caffe'
DISPLAY_CAFFE2 = 'Caffe2'
DISPLAY_PYTORCH = 'Pytorch'
DISPLAY_BLUECONNECT = 'Blueconnect'
DISPLAY_TORCH = 'Torch'
DISPLAY_MXNET = 'Mxnet'
DISPLAY_THEANO = 'Theano'
DISPLAY_DARKNET = 'Darknet'