tensorflow-helpers/classifier-utils.d.ts

Helper functions to use tensorflow in nodejs for transfer learning, image classification, and more

import * as tf from '@tensorflow/tfjs';
export type ClassifierModelSpec = {
    embeddingFeatures: number;
    hiddenLayers?: number[];
    classes: number;
    classNames?: string[];
};
export declare function createImageClassifier(spec: ClassifierModelSpec): tf.Sequential & {
    classNames?: string[];
};
export type ClassificationResult = {
    label: string;
    /** @description between 0 to 1 */
    confidence: number;
};
export declare function getClassCount(shape: tf.Shape | tf.Shape[]): number;
export declare function topClassifyResult(items: ClassificationResult[]): ClassificationResult;
/**
 * @description the values is returned as is.
 * It should has be applied softmax already.
 * */
export declare function mapWithClassName(classNames: string[], values: ArrayLike<number>, options?: {
    sort?: boolean;
}): ClassificationResult[];
export declare function calcClassWeight(options: {
    classes: number;
    classCounts: number[];
}): number[];
export type IOHandler = Exclude<Parameters<tf.GraphModel['save']>[0], string>;
export type ModelArtifacts = Parameters<Required<IOHandler>['save']>[0];
export type SaveResult = ReturnType<Required<IOHandler>['save']>;
export type ModelArtifactsWithClassNames = ModelArtifacts & {
    classNames?: string[];
};
export declare function checkClassNames(modelArtifact: {
    classNames?: string[];
}, classNames: undefined | string[]): undefined | string[];
export declare function attachClassNames<Model extends object>(model: Model, classNames: undefined | string[]): Model & {
    classNames?: string[];
};