ModelNameMappings Classe

ClassNameToCustomerFacingModelMapClassification = {'AveragedPerceptronMulticlassClassifier': 'AveragedPerceptronClassifier', 'DecisionTreeClassifier': 'DecisionTree', 'ExtraTreesClassifier': 'ExtremeRandomTrees', 'GradientBoostingClassifier': 'GradientBoosting', 'KNeighborsClassifier': 'KNN', 'LightGBMClassifier': 'LightGBM', 'LinearSVMWrapper': 'LinearSVM', 'LogisticRegression': 'LogisticRegression', 'NBWrapper': 'BernoulliNaiveBayes', 'NimbusMlAveragedPerceptronClassifier': 'AveragedPerceptronClassifier', 'RandomForestClassifier': 'RandomForest', 'SGDClassifierWrapper': 'SGD', 'SVCWrapper': 'SVM', 'TFDNNClassifierWrapper': 'TensorFlowDNN', 'TFLinearClassifierWrapper': 'TensorFlowLinearClassifier', 'TabnetClassifier': 'TabnetClassifier', 'XGBoostClassifier': 'XGBoostClassifier'}

ClassNameToCustomerFacingModelMapForecasting = {'Arimax': 'Arimax', 'AutoArima': 'AutoArima', 'Average': 'Average', 'DecisionTreeRegressor': 'DecisionTree', 'ElasticNet': 'ElasticNet', 'ExponentialSmoothing': 'ExponentialSmoothing', 'ExtraTreesRegressor': 'ExtremeRandomTrees', 'GradientBoostingRegressor': 'GradientBoosting', 'KNeighborsRegressor': 'KNN', 'LassoLars': 'LassoLars', 'LightGBMRegressor': 'LightGBM', 'Naive': 'Naive', 'Prophet': 'Prophet', 'RandomForestRegressor': 'RandomForest', 'SGDRegressor': 'SGD', 'SeasonalAverage': 'SeasonalAverage', 'SeasonalNaive': 'SeasonalNaive', 'TCNForecaster': 'TCNForecaster', 'TFDNNRegressorWrapper': 'TensorFlowDNN', 'TFLinearRegressorWrapper': 'TensorFlowLinearRegressor', 'TabnetRegressor': 'TabnetRegressor', 'XGBoostRegressor': 'XGBoostRegressor'}

ClassNameToCustomerFacingModelMapRegression = {'DecisionTreeRegressor': 'DecisionTree', 'ElasticNet': 'ElasticNet', 'ExtraTreesRegressor': 'ExtremeRandomTrees', 'GradientBoostingRegressor': 'GradientBoosting', 'KNeighborsRegressor': 'KNN', 'LassoLars': 'LassoLars', 'LightGBMRegressor': 'LightGBM', 'NimbusMlFastLinearRegressor': 'FastLinearRegressor', 'NimbusMlOnlineGradientDescentRegressor': 'OnlineGradientDescentRegressor', 'RandomForestRegressor': 'RandomForest', 'SGDRegressor': 'SGD', 'TFDNNRegressorWrapper': 'TensorFlowDNN', 'TFLinearRegressorWrapper': 'TensorFlowLinearRegressor', 'TabnetRegressor': 'TabnetRegressor', 'XGBoostRegressor': 'XGBoostRegressor'}

CustomerFacingModelToClassNameModelMapClassification = {'AveragedPerceptronClassifier': 'AveragedPerceptronMulticlassClassifier', 'BernoulliNaiveBayes': 'NBWrapper', 'DecisionTree': 'DecisionTreeClassifier', 'ExtremeRandomTrees': 'ExtraTreesClassifier', 'GradientBoosting': 'GradientBoostingClassifier', 'KNN': 'KNeighborsClassifier', 'LightGBM': 'LightGBMClassifier', 'LinearSVM': 'LinearSVMWrapper', 'LogisticRegression': 'LogisticRegression', 'MultinomialNaiveBayes': 'NBWrapper', 'RandomForest': 'RandomForestClassifier', 'SGD': 'SGDClassifierWrapper', 'SVM': 'SVCWrapper', 'TabnetClassifier': 'TabnetClassifier', 'TensorFlowDNN': 'TFDNNClassifierWrapper', 'TensorFlowLinearClassifier': 'TFLinearClassifierWrapper', 'XGBoostClassifier': 'XGBoostClassifier'}

CustomerFacingModelToClassNameModelMapForecasting = {'Arimax': 'Arimax', 'AutoArima': 'AutoArima', 'Average': 'Average', 'DecisionTree': 'DecisionTreeRegressor', 'ElasticNet': 'ElasticNet', 'ExponentialSmoothing': 'ExponentialSmoothing', 'ExtremeRandomTrees': 'ExtraTreesRegressor', 'GradientBoosting': 'GradientBoostingRegressor', 'KNN': 'KNeighborsRegressor', 'LassoLars': 'LassoLars', 'LightGBM': 'LightGBMRegressor', 'Naive': 'Naive', 'Prophet': 'Prophet', 'RandomForest': 'RandomForestRegressor', 'SGD': 'SGDRegressor', 'SeasonalAverage': 'SeasonalAverage', 'SeasonalNaive': 'SeasonalNaive', 'TCNForecaster': 'TCNForecaster', 'TabnetRegressor': 'TabnetRegressor', 'TensorFlowDNN': 'TFDNNRegressorWrapper', 'TensorFlowLinearRegressor': 'TFLinearRegressorWrapper', 'XGBoostRegressor': 'XGBoostRegressor'}

CustomerFacingModelToClassNameModelMapRegression = {'DecisionTree': 'DecisionTreeRegressor', 'ElasticNet': 'ElasticNet', 'ExtremeRandomTrees': 'ExtraTreesRegressor', 'FastLinearRegressor': 'NimbusMlFastLinearRegressor', 'GradientBoosting': 'GradientBoostingRegressor', 'KNN': 'KNeighborsRegressor', 'LassoLars': 'LassoLars', 'LightGBM': 'LightGBMRegressor', 'OnlineGradientDescentRegressor': 'NimbusMlOnlineGradientDescentRegressor', 'RandomForest': 'RandomForestRegressor', 'SGD': 'SGDRegressor', 'TabnetRegressor': 'TabnetRegressor', 'TensorFlowDNN': 'TFDNNRegressorWrapper', 'TensorFlowLinearRegressor': 'TFLinearRegressorWrapper', 'XGBoostRegressor': 'XGBoostRegressor'}

CustomerFacingModelToLegacyModelMapClassification = {'AveragedPerceptronClassifier': 'averaged perceptron classifier', 'BernoulliNaiveBayes': 'BernoulliNB', 'DecisionTree': 'DT', 'ExtremeRandomTrees': 'extra trees', 'GradientBoosting': 'gradient boosting', 'KNN': 'kNN', 'LightGBM': 'lgbm_classifier', 'LinearSVM': 'LinearSVM', 'LogisticRegression': 'logistic regression', 'MultinomialNaiveBayes': 'MultinomialNB', 'RandomForest': 'RF', 'SGD': 'SGD classifier', 'SVM': 'SVM', 'TabnetClassifier': 'tabnet_classifier', 'TensorFlowDNN': 'TF DNNClassifier', 'TensorFlowLinearClassifier': 'TF LinearClassifier', 'XGBoostClassifier': 'xgboost classifier'}

CustomerFacingModelToLegacyModelMapForecasting = {'Arimax': None, 'AutoArima': None, 'Average': None, 'DecisionTree': 'DT regressor', 'ElasticNet': 'Elastic net', 'ExponentialSmoothing': None, 'ExtremeRandomTrees': 'extra trees regressor', 'GradientBoosting': 'Gradient boosting regressor', 'KNN': 'kNN regressor', 'LassoLars': 'Lasso lars', 'LightGBM': 'lightGBM regressor', 'Naive': None, 'Prophet': None, 'RandomForest': 'RF regressor', 'SGD': 'SGD regressor', 'SeasonalAverage': None, 'SeasonalNaive': None, 'TCNForecaster': None, 'TabnetRegressor': 'tabnet_regressor', 'TensorFlowDNN': 'TF DNNRegressor', 'TensorFlowLinearRegressor': 'TF LinearRegressor', 'XGBoostRegressor': 'xgboost regressor'}

CustomerFacingModelToLegacyModelMapRegression = {'DecisionTree': 'DT regressor', 'ElasticNet': 'Elastic net', 'ExtremeRandomTrees': 'extra trees regressor', 'FastLinearRegressor': 'nimbusml fast linear regressor', 'GradientBoosting': 'Gradient boosting regressor', 'KNN': 'kNN regressor', 'LassoLars': 'Lasso lars', 'LightGBM': 'lightGBM regressor', 'OnlineGradientDescentRegressor': 'nimbusml online gradient descent regressor', 'RandomForest': 'RF regressor', 'SGD': 'SGD regressor', 'TabnetRegressor': 'tabnet_regressor', 'TensorFlowDNN': 'TF DNNRegressor', 'TensorFlowLinearRegressor': 'TF LinearRegressor', 'XGBoostRegressor': 'xgboost regressor'}

PrivateModelToLegacyModelMapClassification = {}

PrivateModelToLegacyModelMapForecasting = {}

PrivateModelToLegacyModelMapRegression = {}