tensorflow-helpers/internal.d.ts

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

import { ModelArtifacts, ModelJSON } from '@tensorflow/tfjs-core/dist/io/types';
import { PatchedModelArtifacts } from './model-artifacts';
export type SavedModelJSON = PatchedModelArtifacts & {
    /** @deprecated use userDefinedMetadata.classNames instead */
    classNames?: string[];
};
export declare function patchLoadedModelJSON(model: SavedModelJSON): boolean;
/**
 * @deprecated use ModelJSON.userDefinedMetadata.classNames instead
 */
export type ModelArtifactsWithClassNames = ModelArtifacts & {
    classNames?: string[];
};
export declare function checkClassNames(modelArtifact: ModelJSON, classNames: undefined | string[]): undefined | string[];
type AnyModel = {
    save: Function;
};
export declare function attachClassNames<Model extends AnyModel>(_model: Model, classNames: undefined | string[]): Model & {
    getArtifacts: () => PatchedModelArtifacts;
    classNames?: string[];
};
export {};