learning_model 1.0.50 → 1.0.52

package/dist/index.bundle.js.LICENSE.txt

@@ -1,335 +0,0 @@
-/**
- * @license
- * Copyright 2017 Google LLC. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2018 Google LLC
- *
- * Use of this source code is governed by an MIT-style
- * license that can be found in the LICENSE file or at
- * https://opensource.org/licenses/MIT.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2018 Google LLC. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- *
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2018 Google LLC. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2019 Google LLC
- *
- * Use of this source code is governed by an MIT-style
- * license that can be found in the LICENSE file or at
- * https://opensource.org/licenses/MIT.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2019 Google LLC. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- *
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2019 Google LLC. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2020 Google Inc. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2020 Google LLC
- *
- * Use of this source code is governed by an MIT-style
- * license that can be found in the LICENSE file or at
- * https://opensource.org/licenses/MIT.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2020 Google LLC. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2020 Google LLC. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the License);
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an AS IS BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2021 Google LLC. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2021 Google LLC. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * https://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2022 CodeSmith LLC
- *
- * Use of this source code is governed by an MIT-style
- * license that can be found in the LICENSE file or at
- * https://opensource.org/licenses/MIT.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2022 Google Inc. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2022 Google LLC
- *
- * Use of this source code is governed by an MIT-style
- * license that can be found in the LICENSE file or at
- * https://opensource.org/licenses/MIT.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2022 Google LLC.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2022 Google LLC. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2022 Google LLC. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the 'License');
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an 'AS IS' BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2023 CodeSmith LLC
- *
- * Use of this source code is governed by an MIT-style
- * license that can be found in the LICENSE file or at
- * https://opensource.org/licenses/MIT.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2023 Google LLC.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/**
- * @license
- * Copyright 2023 Google LLC. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-
-/** @license See the LICENSE file. */

package/dist/index.d.ts

@@ -1,3 +0,0 @@
-import LearningMobilenet from './learning/mobilenet';
-import DataModel from './learning/data_model';
-export { LearningMobilenet, DataModel };

package/dist/index.html

@@ -1 +0,0 @@
-<!doctype html><html><head><link rel="stylesheet" href="index.css"><title>My CNN Module</title><script defer="defer" src="index.bundle.js"></script></head><body><div id="loss-bar"></div><div id="progress-bar"></div><div id="root"></div><div id="container"><canvas id="canvas"></canvas></div></body></html>

package/dist/index.js

@@ -1,10 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.DataModel = exports.LearningMobilenet = void 0;
-const mobilenet_1 = __importDefault(require("./learning/mobilenet"));
-exports.LearningMobilenet = mobilenet_1.default;
-const data_model_1 = __importDefault(require("./learning/data_model"));
-exports.DataModel = data_model_1.default;

package/dist/learning/base.d.ts

@@ -1,23 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-import { io } from '@tensorflow/tfjs-core';
-interface LearningInterface {
-    model: tf.LayersModel | null;
-    isRunning: boolean;
-    isReady: boolean;
-    onProgress(progress: number): void;
-    onLoss(loss: number): void;
-    onTrainBegin(log: any): void;
-    onTrainEnd(log: any): void;
-    onEpochEnd(epoch: number, logs: any): void;
-    addData(label: string, data: any): Promise<void>;
-    train(): Promise<tf.History>;
-    infer(data: any): Promise<any>;
-    saveModel(handlerOrURL: io.IOHandler | string, config?: io.SaveConfig): Promise<void>;
-    running(): boolean;
-    ready(): boolean;
-    load({ jsonURL, labels }: {
-        jsonURL: string;
-        labels: Array<string>;
-    }): Promise<void>;
-}
-export default LearningInterface;

package/dist/learning/base.js

@@ -1,2 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });

package/dist/learning/data_model.d.ts

@@ -1,41 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-import { io } from '@tensorflow/tfjs-core';
-import LearningInterface from './base';
-import DataManager from '../utils/data_manager';
-declare class DataModel implements LearningInterface {
-    epochs: number;
-    batchSize: number;
-    learningRate: number;
-    validateRate: number;
-    isTrainedDone: boolean;
-    limitSize: number;
-    isRunning: boolean;
-    isReady: boolean;
-    model: tf.LayersModel | null;
-    dataManager: DataManager;
-    constructor({ epochs, batchSize, limitSize, learningRate, validateRate, }?: {
-        epochs?: number;
-        batchSize?: number;
-        limitSize?: number;
-        learningRate?: number;
-        validateRate?: number;
-    });
-    onProgress(progress: number): void;
-    onLoss(loss: number): void;
-    onTrainBegin(log: any): void;
-    onTrainEnd(log: any): void;
-    onEpochEnd(epoch: number, logs: any): void;
-    addData(label: string, data: number[]): Promise<void>;
-    augmentData(data: number[]): number[];
-    normalizeData(data: number[]): number[];
-    train(): Promise<tf.History>;
-    infer(data: number[]): Promise<Map<string, number>>;
-    saveModel(handlerOrURL: io.IOHandler | string, config?: io.SaveConfig): Promise<void>;
-    running(): boolean;
-    ready(): boolean;
-    load({ jsonURL, labels }: {
-        jsonURL: string;
-        labels: Array<string>;
-    }): Promise<void>;
-}
-export default DataModel;

package/dist/learning/data_model.js

@@ -1,205 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || function (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
-};
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const tf = __importStar(require("@tensorflow/tfjs"));
-const data_manager_1 = __importDefault(require("../utils/data_manager"));
-class DataModel {
-    constructor({ epochs = 50, batchSize = 16, limitSize = 2, learningRate = 0.001, validateRate = 0.15, } = {}) {
-        this.model = null;
-        this.dataManager = new data_manager_1.default();
-        this.epochs = epochs;
-        this.batchSize = batchSize;
-        this.learningRate = learningRate;
-        this.validateRate = validateRate;
-        this.isRunning = false;
-        this.isReady = false;
-        this.isTrainedDone = false;
-        this.limitSize = limitSize;
-    }
-    onProgress(progress) {
-        console.log(`Training progress: ${progress}%`);
-    }
-    onLoss(loss) {
-        console.log(`Loss: ${loss}`);
-    }
-    onTrainBegin(log) {
-        console.log('Training started', log);
-    }
-    onTrainEnd(log) {
-        console.log('Training ended', log);
-    }
-    onEpochEnd(epoch, logs) {
-        console.log(`Epoch ${epoch} ended with logs:`, logs);
-    }
-    addData(label, data) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const normalizedData = this.normalizeData(data);
-            const augmentedData = this.augmentData(normalizedData);
-            yield this.dataManager.addData(label, augmentedData);
-        });
-    }
-    augmentData(data) {
-        return data.map(value => value + (Math.random() * 0.1 - 0.05)); // 약간의 노이즈 추가
-    }
-    normalizeData(data) {
-        const mean = data.reduce((sum, value) => sum + value, 0) / data.length;
-        const std = Math.sqrt(data.reduce((sum, value) => sum + Math.pow(value - mean, 2), 0) / data.length);
-        return data.map(value => (value - mean) / std);
-    }
-    train() {
-        return __awaiter(this, void 0, void 0, function* () {
-            if (this.isRunning) {
-                throw new Error('Training is already in progress');
-            }
-            const { xs, ys } = this.dataManager.convertToTensors();
-            const jointModel = tf.sequential();
-            jointModel.add(tf.layers.dense({
-                units: 32,
-                activation: 'relu',
-                inputShape: [xs.shape[1]],
-                kernelRegularizer: tf.regularizers.l2({ l2: 0.01 }) // L2 정규화 추가
-            }));
-            jointModel.add(tf.layers.dropout({ rate: 0.5 })); // 드롭아웃 레이어 추가
-            jointModel.add(tf.layers.dense({
-                units: 16,
-                activation: 'relu',
-                kernelRegularizer: tf.regularizers.l2({ l2: 0.01 }) // L2 정규화 추가
-            }));
-            jointModel.add(tf.layers.dropout({ rate: 0.5 })); // 드롭아웃 레이어 추가
-            jointModel.add(tf.layers.dense({ units: Object.keys(this.dataManager.getLabelMap()).length, activation: 'softmax' }));
-            jointModel.compile({
-                optimizer: tf.train.adam(this.learningRate),
-                loss: 'sparseCategoricalCrossentropy',
-                metrics: ['accuracy']
-            });
-            this.model = jointModel;
-            this.isRunning = true;
-            const history = yield jointModel.fit(xs, ys, {
-                epochs: this.epochs,
-                batchSize: this.batchSize,
-                validationSplit: 0.2,
-                callbacks: {
-                    onEpochEnd: (epoch, logs) => this.onEpochEnd(epoch, logs),
-                    onBatchEnd: (batch, logs) => {
-                    },
-                    onTrainBegin: (logs) => this.onTrainBegin(logs),
-                    onTrainEnd: (logs) => {
-                        this.isRunning = false;
-                        this.isReady = true;
-                        this.onTrainEnd(logs);
-                    },
-                }
-            });
-            return history;
-        });
-    }
-    infer(data) {
-        return __awaiter(this, void 0, void 0, function* () {
-            if (!this.isReady || !this.model) {
-                throw new Error('Model is not ready');
-            }
-            // 입력 데이터를 텐서로 변환
-            const input = tf.tensor2d([data], [1, data.length], 'float32');
-            try {
-                // 모델 예측 수행
-                const prediction = this.model.predict(input);
-                // 예측 결과 값들을 추출
-                const values = yield prediction.data();
-                // 클래스 이름과 확률을 저장할 Map 생성
-                const classProbabilities = new Map();
-                const EPSILON = 1e-6; // 매우 작은 값을 표현하기 위한 엡실론
-                // 라벨 맵을 가져옴
-                const labelMap = this.dataManager.getLabelMap();
-                const reverseLabelMap = Object.entries(labelMap).reduce((acc, [key, value]) => {
-                    acc[value] = key;
-                    return acc;
-                }, {});
-                // 각 클래스의 확률 계산 및 조정
-                for (let i = 0; i < values.length; i++) {
-                    let probability = Math.max(0, Math.min(1, values[i])); // 확률 값을 0과 1 사이로 조정
-                    probability = probability < EPSILON ? 0 : probability; // 매우 작은 확률 값을 0으로 간주
-                    const className = reverseLabelMap[i]; // 클래스 이름을 문자열로 가져옴
-                    if (className) {
-                        const existingProbability = classProbabilities.get(className);
-                        if (existingProbability !== undefined) {
-                            classProbabilities.set(className, existingProbability + probability);
-                        }
-                        else {
-                            classProbabilities.set(className, probability);
-                        }
-                    }
-                }
-                // 텐서 해제
-                prediction.dispose();
-                input.dispose();
-                console.log('classProbabilities', classProbabilities);
-                return classProbabilities;
-            }
-            catch (error) {
-                // 텐서 해제
-                input.dispose();
-                throw error;
-            }
-        });
-    }
-    saveModel(handlerOrURL, config) {
-        return __awaiter(this, void 0, void 0, function* () {
-            if (!this.model) {
-                throw new Error('No model to save');
-            }
-            yield this.model.save(handlerOrURL, config);
-        });
-    }
-    running() {
-        return this.isRunning;
-    }
-    ready() {
-        return this.isReady;
-    }
-    load({ jsonURL, labels }) {
-        return __awaiter(this, void 0, void 0, function* () {
-            this.model = yield tf.loadLayersModel(jsonURL);
-            this.isReady = true;
-            // Assuming the labels are provided in the correct order
-            this.dataManager = new data_manager_1.default();
-            labels.forEach((label, index) => {
-                this.dataManager.getLabelMap()[label] = index;
-            });
-        });
-    }
-}
-exports.default = DataModel;

package/dist/learning/data_model.test.d.ts

@@ -1 +0,0 @@
-export {};