learning_model 1.0.17 → 1.0.22

package/src/learning/mobilenet_image.test.ts

@@ -1,63 +0,0 @@
-import * as fs from 'fs';
-import * as path from 'path';
-import * as tf from '@tensorflow/tfjs';
-import LearningMobilenetImage from './mobilenet_image';
-
-describe('LearningMobilenetImage', () => {
-  let learning: LearningMobilenetImage;
-  const imagePath = path.join(__dirname, 'data/images/image.jpg'); // 이미지 파일 경로
-
-  beforeEach(() => {
-    learning = new LearningMobilenetImage({});
-  });
-
-  afterEach(() => {
-    // 테스트 후에 모델 저장
-    learning.saveModel();
-  });
-
-  it('should add data', () => {
-    const label = 'label1';
-    const image = fs.readFileSync(imagePath); // 이미지 파일 읽어오기
-
-    learning.addData(label, image);
-
-    expect(learning.labels).toContain(label);
-    expect(learning.trainImages.length).toBe(1);
-  });
-
-  it('should not train if data is insufficient', async () => {
-    spyOn(console, 'error'); // 콘솔 에러 로그를 가로채기 위해 spyOn 사용
-
-    const result = await learning.train().catch((error: any) => error);
-
-    expect(result).toBeInstanceOf(Error);
-    expect(result.message).toBe('Please train Data need over 2 data length');
-    expect(console.error).toHaveBeenCalledWith('Model training failed', result);
-    expect(learning.model).toBeNull();
-  });
-
-  it('should train and infer', async () => {
-    const label1 = 'label1';
-    const image1 = fs.readFileSync(imagePath); // 이미지 파일 읽어오기
-    const label2 = 'label2';
-    const image2 = fs.readFileSync(imagePath); // 이미지 파일 읽어오기
-    const testData = fs.readFileSync(imagePath); // 테스트 데이터 읽어오기
-
-    learning.addData(label1, image1);
-    learning.addData(label2, image2);
-
-    const history = await learning.train();
-
-    expect(history).toBeDefined();
-    expect(learning.model).toBeInstanceOf(tf.LayersModel);
-    expect(learning.isRunning).toBe(false);
-
-    const predictions = await learning.infer(testData);
-
-    expect(predictions).toBeInstanceOf(Map);
-    expect(predictions.size).toBe(2);
-    // expect(predictions.get(label1)).toBeCloseTo(/* 예측값 */);
-    // expect(predictions.get(label2)).toBeCloseTo(/* 예측값 */);
-  });
-});

package/types/index.d.ts

@@ -1,3 +0,0 @@
-import LearningImage from './learning/image';
-import LearningMobilenetImage from './learning/mobilenet_image';
-export { LearningImage, LearningMobilenetImage };

package/types/learning/base.d.ts

@@ -1,19 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-interface LearningInterface {
-    model: tf.LayersModel | null;
-    labels: string[];
-    isRunning: boolean;
-    isReady: boolean;
-    onProgress(progress: number): void;
-    onLoss(loss: number): void;
-    onEvents(logs: any): void;
-    onTrainBegin(log: any): void;
-    onTrainEnd(log: any): void;
-    addData(label: string, data: any): Promise<void>;
-    train(): Promise<tf.History>;
-    infer(data: any): Promise<any>;
-    saveModel(): void;
-    running(): boolean;
-    ready(): boolean;
-}
-export default LearningInterface;

package/types/learning/image.d.ts

@@ -1,40 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-import LearningInterface from './base';
-declare class LearningImage implements LearningInterface {
-    model: tf.LayersModel | null;
-    epochs: number;
-    batchSize: number;
-    learningRate: number;
-    labels: string[];
-    isRunning: boolean;
-    isReady: boolean;
-    limitSize: number;
-    trainImages: tf.Tensor3D[];
-    readonly MOBILE_NET_INPUT_WIDTH = 224;
-    readonly MOBILE_NET_INPUT_HEIGHT = 224;
-    readonly MOBILE_NET_INPUT_CHANNEL = 3;
-    readonly IMAGE_NORMALIZATION_FACTOR = 255;
-    constructor({ epochs, batchSize, limitSize, learningRate, }?: {
-        modelURL?: string;
-        epochs?: number;
-        batchSize?: number;
-        limitSize?: number;
-        learningRate?: number;
-    });
-    onProgress: (progress: number) => void;
-    onLoss: (loss: number) => void;
-    onEvents: (logs: any) => void;
-    onTrainBegin: (log: any) => void;
-    onTrainEnd: (log: any) => void;
-    addData(label: string, data: any): Promise<void>;
-    train(): Promise<tf.History>;
-    infer(data: any): Promise<Map<string, number>>;
-    saveModel(): void;
-    running(): boolean;
-    ready(): boolean;
-    private _preprocessedTargetData;
-    private _preprocessedInputData;
-    private _preprocessData;
-    private _createModel;
-}
-export default LearningImage;

package/types/learning/mobilenet_image.d.ts

@@ -1,42 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-import LearningInterface from './base';
-declare class LearningMobilenetImage implements LearningInterface {
-    model: tf.LayersModel | null;
-    epochs: number;
-    batchSize: number;
-    learningRate: number;
-    labels: string[];
-    modelURL: string;
-    isRunning: boolean;
-    isReady: boolean;
-    limitSize: number;
-    trainImages: tf.Tensor3D[];
-    readonly MOBILE_NET_INPUT_WIDTH = 224;
-    readonly MOBILE_NET_INPUT_HEIGHT = 224;
-    readonly MOBILE_NET_INPUT_CHANNEL = 3;
-    readonly IMAGE_NORMALIZATION_FACTOR = 255;
-    constructor({ modelURL, // 디폴트 mobilenet 이미지 
-    epochs, batchSize, limitSize, learningRate, }?: {
-        modelURL?: string;
-        epochs?: number;
-        batchSize?: number;
-        limitSize?: number;
-        learningRate?: number;
-    });
-    onProgress: (progress: number) => void;
-    onLoss: (loss: number) => void;
-    onEvents: (logs: any) => void;
-    onTrainBegin: (log: any) => void;
-    onTrainEnd: (log: any) => void;
-    addData(label: string, data: any): Promise<void>;
-    train(): Promise<tf.History>;
-    infer(data: any): Promise<Map<string, number>>;
-    saveModel(): void;
-    running(): boolean;
-    ready(): boolean;
-    private _preprocessedTargetData;
-    private _preprocessedInputData;
-    private _preprocessData;
-    private _createModel;
-}
-export default LearningMobilenetImage;

package/types/learning/mobilenet_image.test.d.ts

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

package/types/public/index.d.ts

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

package/types/src/index.d.ts

@@ -1,3 +0,0 @@
-import LearningImage from './learning/image';
-import LearningMobilenetImage from './learning/mobilenet_image';
-export { LearningImage, LearningMobilenetImage };

package/types/src/learning/base.d.ts

@@ -1,19 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-interface LearningInterface {
-    model: tf.LayersModel | null;
-    labels: string[];
-    isRunning: boolean;
-    isReady: boolean;
-    onProgress(progress: number): void;
-    onLoss(loss: number): void;
-    onEvents(logs: any): void;
-    onTrainBegin(log: any): void;
-    onTrainEnd(log: any): void;
-    addData(label: string, data: any): Promise<void>;
-    train(): Promise<tf.History>;
-    infer(data: any): Promise<any>;
-    saveModel(): void;
-    running(): boolean;
-    ready(): boolean;
-}
-export default LearningInterface;

package/types/src/learning/image.d.ts

@@ -1,40 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-import LearningInterface from './base';
-declare class LearningImage implements LearningInterface {
-    model: tf.LayersModel | null;
-    epochs: number;
-    batchSize: number;
-    learningRate: number;
-    labels: string[];
-    isRunning: boolean;
-    isReady: boolean;
-    limitSize: number;
-    trainImages: tf.Tensor3D[];
-    readonly MOBILE_NET_INPUT_WIDTH = 224;
-    readonly MOBILE_NET_INPUT_HEIGHT = 224;
-    readonly MOBILE_NET_INPUT_CHANNEL = 3;
-    readonly IMAGE_NORMALIZATION_FACTOR = 255;
-    constructor({ epochs, batchSize, limitSize, learningRate, }?: {
-        modelURL?: string;
-        epochs?: number;
-        batchSize?: number;
-        limitSize?: number;
-        learningRate?: number;
-    });
-    onProgress: (progress: number) => void;
-    onLoss: (loss: number) => void;
-    onEvents: (logs: any) => void;
-    onTrainBegin: (log: any) => void;
-    onTrainEnd: (log: any) => void;
-    addData(label: string, data: any): Promise<void>;
-    train(): Promise<tf.History>;
-    infer(data: any): Promise<Map<string, number>>;
-    saveModel(): void;
-    running(): boolean;
-    ready(): boolean;
-    private _preprocessedTargetData;
-    private _preprocessedInputData;
-    private _preprocessData;
-    private _createModel;
-}
-export default LearningImage;

package/types/src/learning/mobilenet_image.d.ts

@@ -1,42 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-import LearningInterface from './base';
-declare class LearningMobilenetImage implements LearningInterface {
-    model: tf.LayersModel | null;
-    epochs: number;
-    batchSize: number;
-    learningRate: number;
-    labels: string[];
-    modelURL: string;
-    isRunning: boolean;
-    isReady: boolean;
-    limitSize: number;
-    trainImages: tf.Tensor3D[];
-    readonly MOBILE_NET_INPUT_WIDTH = 224;
-    readonly MOBILE_NET_INPUT_HEIGHT = 224;
-    readonly MOBILE_NET_INPUT_CHANNEL = 3;
-    readonly IMAGE_NORMALIZATION_FACTOR = 255;
-    constructor({ modelURL, // 디폴트 mobilenet 이미지 
-    epochs, batchSize, limitSize, learningRate, }?: {
-        modelURL?: string;
-        epochs?: number;
-        batchSize?: number;
-        limitSize?: number;
-        learningRate?: number;
-    });
-    onProgress: (progress: number) => void;
-    onLoss: (loss: number) => void;
-    onEvents: (logs: any) => void;
-    onTrainBegin: (log: any) => void;
-    onTrainEnd: (log: any) => void;
-    addData(label: string, data: any): Promise<void>;
-    train(): Promise<tf.History>;
-    infer(data: any): Promise<Map<string, number>>;
-    saveModel(): void;
-    running(): boolean;
-    ready(): boolean;
-    private _preprocessedTargetData;
-    private _preprocessedInputData;
-    private _preprocessData;
-    private _createModel;
-}
-export default LearningMobilenetImage;