@@ -448,6 +448,10 @@ define([
448448 return Config . ML_DATA_TYPES ;
449449 }
450450
451+ getMLCategories ( ) {
452+ return Object . keys ( Config . ML_DATA_DICT ) ;
453+ }
454+
451455 }
452456
453457 //========================================================================
@@ -482,38 +486,38 @@ define([
482486 * Data types using for searching model variables
483487 */
484488 Config . ML_DATA_DICT = {
489+ 'Data Preparation' : [
490+ /** Encoding */
491+ 'OneHotEncoder' , 'LabelEncoder' , 'OrdinalEncoder' , 'TargetEncoder' , 'SMOTE' ,
492+ /** Scaling */
493+ 'StandardScaler' , 'RobustScaler' , 'MinMaxScaler' , 'Normalizer' , 'FunctionTransformer' , 'PolynomialFeatures' , 'KBinsDiscretizer' ,
494+ /** ETC */
495+ 'ColumnTransformer'
496+ ] ,
485497 'Regression' : [
486498 'LinearRegression' , 'Ridge' , 'Lasso' , 'ElasticNet' , 'SVR' , 'DecisionTreeRegressor' , 'RandomForestRegressor' , 'GradientBoostingRegressor' , 'XGBRegressor' , 'LGBMRegressor' , 'CatBoostRegressor' ,
487499 ] ,
488500 'Classification' : [
489501 'LogisticRegression' , 'BernoulliNB' , 'MultinomialNB' , 'GaussianNB' , 'SVC' , 'DecisionTreeClassifier' , 'RandomForestClassifier' , 'GradientBoostingClassifier' , 'XGBClassifier' , 'LGBMClassifier' , 'CatBoostClassifier' ,
490502 ] ,
491- 'Auto ML' : [
492- 'AutoSklearnRegressor' , 'AutoSklearnClassifier' , 'TPOTRegressor' , 'TPOTClassifier'
493- ] ,
494503 'Clustering' : [
495504 'KMeans' , 'AgglomerativeClustering' , 'GaussianMixture' , 'DBSCAN' ,
496505 ] ,
497506 'Dimension Reduction' : [
498507 'PCA' , 'LinearDiscriminantAnalysis' , 'TruncatedSVD' , 'NMF' , 'TSNE'
499508 ] ,
500- 'Data Preparation' : [
501- /** Encoding */
502- 'OneHotEncoder' , 'LabelEncoder' , 'OrdinalEncoder' , 'TargetEncoder' , 'SMOTE' ,
503- /** Scaling */
504- 'StandardScaler' , 'RobustScaler' , 'MinMaxScaler' , 'Normalizer' , 'FunctionTransformer' , 'PolynomialFeatures' , 'KBinsDiscretizer' ,
505- /** ETC */
506- 'ColumnTransformer'
509+ 'Auto ML' : [
510+ 'AutoSklearnRegressor' , 'AutoSklearnClassifier' , 'TPOTRegressor' , 'TPOTClassifier'
507511 ]
508512 } ;
509513
510514 Config . ML_DATA_TYPES = [
515+ ...Config . ML_DATA_DICT [ 'Data Preparation' ] ,
511516 ...Config . ML_DATA_DICT [ 'Regression' ] ,
512517 ...Config . ML_DATA_DICT [ 'Classification' ] ,
513- ...Config . ML_DATA_DICT [ 'Auto ML' ] ,
514518 ...Config . ML_DATA_DICT [ 'Clustering' ] ,
515519 ...Config . ML_DATA_DICT [ 'Dimension Reduction' ] ,
516- ...Config . ML_DATA_DICT [ 'Data Preparation ' ]
520+ ...Config . ML_DATA_DICT [ 'Auto ML ' ]
517521 ] ;
518522
519523 return Config ;
0 commit comments