add Neural Network as default classifier to UI

Author: mkaliberda

src/ui/src/components/ModelSummary/ModelSummary.jsx

@@ -22,6 +22,7 @@ import React, { useState, useEffect } from "react";
 import { Box, Typography } from "@mui/material";
 import StandardTable from "components/StandardTable";
 import { ColumnType } from "components/StandardTable/StandardTableConstants";
+import { NN_CLASSIFIER_NAME_LIST } from "consts";
 import NeuralNetworkSummary from "./NeuralNetworkSummary";
 import PMEModelSummary from "./PMEModelSummary";
 
@@ -72,7 +73,7 @@ const ModelSummary = ({ model, showTitle }) => {
           m.pipeline_summary[pipelineSummaryCount - 1].optimizers[0].inputs.max_depth;
       } else if (m.device_configuration?.classifier === "Boosted Tree Ensemble") {
         //
-      } else if (m.device_configuration?.classifier === "TensorFlow Lite for Microcontrollers") {
+      } else if (NN_CLASSIFIER_NAME_LIST.includes(m.device_configuration?.classifier)) {
         //
       }
     }
@@ -131,7 +132,7 @@ const ModelSummary = ({ model, showTitle }) => {
       {model?.data?.device_configuration?.classifier === "PME" && modelParameters ? (
         <PMEModelSummary model={modelParameters} />
       ) : null}
-      {model?.data?.device_configuration?.classifier === "TensorFlow Lite for Microcontrollers" ? (
+      {NN_CLASSIFIER_NAME_LIST.includes(model?.data?.device_configuration?.classifier) ? (
         <NeuralNetworkSummary model={model} />
       ) : null}
     </Box>

src/ui/src/components/PipelineCreateForm/PipelineCreateForm.jsx

@@ -20,6 +20,7 @@ License along with SensiML Piccolo AI. If not, see <https://www.gnu.org/licenses
 import React, { useState } from "react";
 import { Box, Stepper, Step, StepLabel } from "@mui/material";
 import { useTranslation } from "react-i18next";
+import { DEFAULT_CLASSIFIER } from "consts";
 
 import PipelineQueryCreateForm from "components/PipelineQueryCreateForm";
 
@@ -75,7 +76,11 @@ const PipelineCreateForm = ({
         ))}
       </Stepper>
       {activeStep === STEP_PARAMS ? (
-        <PipelineCreateFormParameters classifiers={classifiers} onSubmit={handleSubmitParams} />
+        <PipelineCreateFormParameters
+          classifiers={classifiers}
+          defaultClassifier={DEFAULT_CLASSIFIER}
+          onSubmit={handleSubmitParams}
+        />
       ) : (
         <PipelineQueryCreateForm
           pipelineError={pipelineError}

src/ui/src/consts/CLASSIFIERS.js

@@ -0,0 +1,3 @@
+export const NN_CLASSIFIER_NAME_LIST = ["TensorFlow Lite for Microcontrollers", "Neural Network"];
+export const DISABLED_CLASSIFIERS = ["TF Micro", "TensorFlow Lite for Microcontrollers"];
+export const DEFAULT_CLASSIFIER = "Neural Network";

src/ui/src/consts/index.js

@@ -24,6 +24,7 @@ import CONSOLE from "./CONSOLE";
 import RESPONSIVE from "./RESPONSIVE";
 import PIPELINE_CONSTS from "./PIPELINE_CONSTS";
 
+export * from "./CLASSIFIERS";
 export * from "./CHARTS_TYPES";
 
 export {

src/ui/src/containers/DownloadModel/components/TargetDeviceOptions.jsx

@@ -41,6 +41,7 @@ import Autocomplete from "@mui/material/Autocomplete";
 import FormCurtain from "components/UIFormCurtain";
 import CheckBoxSelectCard from "components/FormElements/CheckBoxSelectCard";
 
+import { NN_CLASSIFIER_NAME_LIST } from "consts";
 import {
   PLATROFM_TYPES,
   STEP_SELECT_PLATFORM,
@@ -56,8 +57,6 @@ import useStyles from "../DownloadModelStyles";
 import AdvanceSettings from "./AdvanceSettings";
 import defaultSelectedPlatform from "./defaultSelectedPlatform";
 
-const TF_CLASSIFIER_NAME = "TensorFlow Lite for Microcontrollers";
-
 /*
 const defaultEmpty = {
   target_platform: "",
@@ -180,7 +179,7 @@ const TargetDeviceOptions = forwardRef(
         ),
         target_compiler: platform?.default_selections?.compiler || "",
       };
-      if (platform?.nn_inference_engines && classifierName === TF_CLASSIFIER_NAME) {
+      if (platform?.nn_inference_engines && NN_CLASSIFIER_NAME_LIST.includes(classifierName)) {
         const nnList = _.keys(platform.nn_inference_engines);
         if (!_.isEmpty(nnList)) {
           // eslint-disable-next-line prefer-destructuring

src/ui/src/store/autoML/const.js

@@ -87,7 +87,7 @@ export const TVO_LIST = [
   PIPELINE_STEP_TYPES.VALIDATION,
 ];
 
-export const DISABLED_CLASSIFIERS = ["TF Micro"];
+export const DISABLED_CLASSIFIERS = ["TF Micro", "TensorFlow Lite for Microcontrollers"];
 
 export const ONETIME_ADDED = [FEATURE_GENERATOR, FEATURE_SELECTOR];
 

src/ui/src/store/pipelineTemplates/data/KWSpottingData.json

@@ -123,7 +123,7 @@
           ],
           "classifiers": [
             {
-              "name": "TensorFlow Lite for Microcontrollers",
+              "name": "Neural Network",
               "inputs": {}
             }
           ],
@@ -296,7 +296,7 @@
           ],
           "classifiers": [
             {
-              "name": "TensorFlow Lite for Microcontrollers",
+              "name": "Neural Network",
               "inputs": {}
             }
           ],
@@ -501,7 +501,7 @@
           ],
           "classifiers": [
             {
-              "name": "TensorFlow Lite for Microcontrollers",
+              "name": "Neural Network",
               "inputs": {}
             }
           ],
@@ -659,7 +659,7 @@
           ],
           "classifiers": [
             {
-              "name": "TensorFlow Lite for Microcontrollers",
+              "name": "Neural Network",
               "inputs": {}
             }
           ],

src/ui/src/store/pipelineTemplates/data/audioData.json

@@ -168,7 +168,7 @@
           ],
           "classifiers": [
             {
-              "name": "TensorFlow Lite for Microcontrollers",
+              "name": "Neural Network",
               "inputs": {}
             }
           ],

src/ui/src/store/transforms/selectors.js

@@ -17,7 +17,7 @@ You should have received a copy of the GNU Affero General Public
 License along with SensiML Piccolo AI. If not, see <https://www.gnu.org/licenses/>.
 */
 
-import { DISABLED_CLASSIFIERS } from "store/autoML/const";
+import { DISABLED_CLASSIFIERS } from "consts";
 
 export const selectTransformsByTypeSubType =
   (type, subtype, transformList, excludeTransform) => (state) => {