@debut/plugin-neurofilter
Version:
Neuro filter plugin for debute
25 lines (24 loc) • 799 B
TypeScript
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;
}