Feedback from comments

automl · Feb 22, 2021 · e336373 · e336373
1 parent 6c026c7
commit e336373
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/autoPyTorch/pipeline/components/setup/traditional_ml/classifier_models/classifiers.py b/autoPyTorch/pipeline/components/setup/traditional_ml/classifier_models/classifiers.py
@@ -110,7 +110,7 @@ def predict(self, X_test: np.ndarray, predict_proba: bool = False) -> np.ndarray
     def get_properties(dataset_properties: Optional[Dict[str, Any]] = None) -> Dict[str, str]:
         return {
             'shortname': 'LGBMClassifier',
-            'name': 'LGBMClassifier',
+            'name': 'Light Gradient Boosting Machine Classifier',
         }
 
 
@@ -178,7 +178,7 @@ def predict(self, X_test: np.ndarray, predict_proba: bool = False) -> np.ndarray
     def get_properties(dataset_properties: Optional[Dict[str, Any]] = None) -> Dict[str, str]:
         return {
             'shortname': 'CatBoostClassifier',
-            'name': 'CatBoostClassifier',
+            'name': 'Categorical Boosting Classifier',
         }
 
 
@@ -243,8 +243,8 @@ def predict(self, X_test: np.ndarray, predict_proba: bool = False) -> np.ndarray
     @staticmethod
     def get_properties(dataset_properties: Optional[Dict[str, Any]] = None) -> Dict[str, str]:
         return {
-            'shortname': 'RandomForestClassifier',
-            'name': 'RandomForestClassifier',
+            'shortname': 'RFClassifier',
+            'name': 'Random Forest Classifier',
         }
 
 
@@ -369,8 +369,8 @@ def predict(self, X_test: np.ndarray, predict_proba: bool = False) -> np.ndarray
     @staticmethod
     def get_properties(dataset_properties: Optional[Dict[str, Any]] = None) -> Dict[str, str]:
         return {
-            'shortname': 'KNeighborsClassifier',
-            'name': 'KNeighborsClassifier',
+            'shortname': 'KNNClassifier',
+            'name': 'K Nearest Neighbors Classifier',
         }
 
 
@@ -424,5 +424,5 @@ def predict(self, X_test: np.ndarray, predict_proba: bool = False) -> np.ndarray
     def get_properties(dataset_properties: Optional[Dict[str, Any]] = None) -> Dict[str, str]:
         return {
             'shortname': 'SVC',
-            'name': 'SVC',
+            'name': 'Support Vector Classification',
         }