import * as tf from '@tensorflow/tfjs';
import { io } from '@tensorflow/tfjs-core';

// AI Learning 디폴트 인터페이스
interface LearningInterface {
    // tf 모델
    model: tf.LayersModel | null;
    // 이미 train중인지 여부 - 이미 진행중인경우 취소
    isRunning: boolean;
    // 준비 
    isReady: boolean;
    
    // 진행상태 이벤트
    onProgress(progress: number): void;

    // loss 정보
    onLoss(loss: number): void;

    // 학습 시작 이벤트
    onTrainBegin(log: any): void;

    // 학습 종료 이벤트
    onTrainEnd(log: any): void;

    onEpochEnd(epoch: number, logs: any): void

    // 학습 데이타 추가 data는 다른 클래스에 맞게 any로 설정
    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;