UNPKG

@debut/plugin-neurofilter

Version:

Neuro filter plugin for debute

25 lines (24 loc) 799 B
import { NeuralNetwork } from 'brain.js'; import { utils } from '@debut/community-core'; export declare type TrainingData<T> = { input: T[]; output: [number] | null; id: string; }; export declare class NeuroHelper<T = number> { trained: boolean; protected nn: NeuralNetwork; protected trainingSet: Array<TrainingData<T>>; protected botData: utils.cli.BotData; constructor(botName: string, nnOptions?: ConstructorParameters<typeof NeuralNetwork>[0]); addTrainingData(data: TrainingData<T>): void; updateTrainingOut(id: string, output: number): void; train(options?: { iterations?: number; errorThresh?: number; log?: boolean; }): void; run(input: T[]): any; save(ticker: string): void; load(ticker: string): void; }