ml-basic
Version:
Lightweight, zero dependency, machine learning library
20 lines (19 loc) • 581 B
TypeScript
import { Activator } from "../lib/functions";
import Matrix from "../lib/matrix";
import Layer from "./layer";
export type FullyConnectedParams = {
input: number | [number, number];
output: number;
/**
* @default {@link Sigmoid}
*/
activation?: Activator;
};
export default class FullyConnectedLayer extends Layer {
name: string;
weights: Matrix;
bias: Matrix;
constructor({ input, output, activation }: FullyConnectedParams);
propagate(input: Matrix): Matrix;
backPropagate(input: Matrix, output: Matrix, loss: Matrix): Matrix;
}