Add info option for Clustering

minjk-bl · minjk-bl · commit 899eb08fc8a1 · 2022-04-26T12:33:51.000+09:00
diff --git a/js/m_ml/ModelInfo.js b/js/m_ml/ModelInfo.js
@@ -551,16 +551,18 @@ define([
                     break;
                 case 'Clustering':
                     infos = {
-                        // 'Size of clusters': {
-                        //     name: 'Size of clusters',
-                        //     code: "print(f'Size of clusters: {np.bincount(pred)}')", // FIXME: model.cluster_centers_ / use model info or hide it
-                        //     options: []
-                        // }
+                        'get_params': {
+                            ...defaultInfos['get_params']
+                        }
                     }
 
                     if (modelType == 'KMeans') {
                         infos = {
                             ...infos,
+                            'score': {
+                                ...defaultInfos['score'],
+                                description: 'Return the mean accuracy on the given test data and labels.'
+                            },
                             'cluster_centers_': {
                                 name: 'cluster_centers',
                                 label: 'Cluster centers',
@@ -587,6 +589,16 @@ define([
                             }
                         }
                     }
+
+                    if (modelType == 'GaussianMixture') {
+                        infos = {
+                            ...infos,
+                            'score': {
+                                ...defaultInfos['score'],
+                                description: 'Compute the per-sample average log-likelihood of the given data X.'
+                            }
+                        }
+                    }
                     break;
                 case 'Dimension Reduction':
                     if (modelType == 'LDA') {
