learning_model 1.0.39 → 1.0.41

package/dist/learning/mobilenet.js

@@ -190,13 +190,13 @@ class LearningMobilenet {
                     this.isRunning = false;
                 },
                 onBatchBegin: (batch, logs) => {
-                    console.log(`Batch ${batch} is starting.`);
+                    //console.log(`Batch ${batch} is starting.`);
                 },
                 onBatchEnd: (batch, logs) => {
-                    console.log(`Batch ${batch} has ended.`);
+                    //console.log(`Batch ${batch} has ended.`);
                 },
                 onEpochBegin: (epoch, logs) => {
-                    console.log(`Epoch ${epoch + 1} is starting.`, logs);
+                    //console.log(`Epoch ${epoch+1} is starting.`, logs);
                 },
                 onEpochEnd: (epoch, logs) => {
                     this.onEpochEnd(epoch, logs);
@@ -264,6 +264,7 @@ class LearningMobilenet {
                         classProbabilities.set(className, probability);
                     }
                 }
+                console.log('classProbabilities', classProbabilities);
                 (0, tfjs_1.dispose)(logits);
                 return classProbabilities;
             }
@@ -300,7 +301,7 @@ class LearningMobilenet {
                     layers: [
                         tf.layers.dense({
                             inputShape: this.mobilenetModule.outputs[0].shape.slice(1),
-                            units: 64,
+                            units: 128,
                             activation: 'relu',
                             kernelInitializer: varianceScaling,
                             useBias: true

package/lib/learning/mobilenet.ts

@@ -199,13 +199,13 @@ class LearningMobilenet implements LearningInterface {
         this.isRunning = false;
       },
       onBatchBegin: (batch: any, logs: any) => {
-        console.log(`Batch ${batch} is starting.`);
+        //console.log(`Batch ${batch} is starting.`);
       },
       onBatchEnd: (batch: any, logs: any) => {
-        console.log(`Batch ${batch} has ended.`);
+        //console.log(`Batch ${batch} has ended.`);
       },
       onEpochBegin: (epoch: number, logs: any) => {
-        console.log(`Epoch ${epoch+1} is starting.`, logs);
+        //console.log(`Epoch ${epoch+1} is starting.`, logs);
       },
       onEpochEnd: (epoch: number, logs: any) => {
         this.onEpochEnd(epoch, logs);
@@ -273,6 +273,7 @@ class LearningMobilenet implements LearningInterface {
           classProbabilities.set(className, probability);
         }
       }
+      console.log('classProbabilities', classProbabilities);
       dispose(logits);
       return classProbabilities;
     } catch (error) {
@@ -310,7 +311,7 @@ class LearningMobilenet implements LearningInterface {
         layers: [
           tf.layers.dense({
             inputShape: this.mobilenetModule!.outputs[0].shape.slice(1),
-            units: 64,
+            units: 128,
             activation: 'relu',
             kernelInitializer: varianceScaling, // 'varianceScaling'
             useBias: true

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "learning_model",
-  "version": "1.0.39",
+  "version": "1.0.41",
   "description": "learning model develop",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",