Skip to content

Commit 3c3797c

Browse files
Feedback from comments
1 parent 8d868e0 commit 3c3797c

File tree

1 file changed

+7
-7
lines changed
  • autoPyTorch/pipeline/components/setup/traditional_ml/classifier_models

1 file changed

+7
-7
lines changed

autoPyTorch/pipeline/components/setup/traditional_ml/classifier_models/classifiers.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ def predict(self, X_test: np.ndarray, predict_proba: bool = False) -> np.ndarray
110110
def get_properties(dataset_properties: Optional[Dict[str, Any]] = None) -> Dict[str, str]:
111111
return {
112112
'shortname': 'LGBMClassifier',
113-
'name': 'LGBMClassifier',
113+
'name': 'Light Gradient Boosting Machine Classifier',
114114
}
115115

116116

@@ -178,7 +178,7 @@ def predict(self, X_test: np.ndarray, predict_proba: bool = False) -> np.ndarray
178178
def get_properties(dataset_properties: Optional[Dict[str, Any]] = None) -> Dict[str, str]:
179179
return {
180180
'shortname': 'CatBoostClassifier',
181-
'name': 'CatBoostClassifier',
181+
'name': 'Categorical Boosting Classifier',
182182
}
183183

184184

@@ -243,8 +243,8 @@ def predict(self, X_test: np.ndarray, predict_proba: bool = False) -> np.ndarray
243243
@staticmethod
244244
def get_properties(dataset_properties: Optional[Dict[str, Any]] = None) -> Dict[str, str]:
245245
return {
246-
'shortname': 'RandomForestClassifier',
247-
'name': 'RandomForestClassifier',
246+
'shortname': 'RFClassifier',
247+
'name': 'Random Forest Classifier',
248248
}
249249

250250

@@ -369,8 +369,8 @@ def predict(self, X_test: np.ndarray, predict_proba: bool = False) -> np.ndarray
369369
@staticmethod
370370
def get_properties(dataset_properties: Optional[Dict[str, Any]] = None) -> Dict[str, str]:
371371
return {
372-
'shortname': 'KNeighborsClassifier',
373-
'name': 'KNeighborsClassifier',
372+
'shortname': 'KNNClassifier',
373+
'name': 'K Nearest Neighbors Classifier',
374374
}
375375

376376

@@ -424,5 +424,5 @@ def predict(self, X_test: np.ndarray, predict_proba: bool = False) -> np.ndarray
424424
def get_properties(dataset_properties: Optional[Dict[str, Any]] = None) -> Dict[str, str]:
425425
return {
426426
'shortname': 'SVC',
427-
'name': 'SVC',
427+
'name': 'Support Vector Classification',
428428
}

0 commit comments

Comments
 (0)