import { Network as INetwork } from "./network";
import { Node } from "./node";
import * as neataptic from "neataptic"

export const Construct: (nodes: Node[]) => INetwork = neataptic.architect.Construct

export const Network: (...sizes: number[]) => INetwork = neataptic.architect.Network

export const LSTM: (...sizes: number[]) => INetwork = neataptic.architect.LSTM

export const GRU: (...sizes: number[]) => INetwork = neataptic.architect.GRU

export const NARX: (
    inputSize: number, 
    hiddenLayers: number, 
    outputSize: number, 
    previousInput: number, 
    previousOutput: number
) => INetwork = neataptic.architect.NARX

export const Perceptron: (...sizes: number[]) => INetwork = neataptic.architect.Perceptron

export type RandomNetworkOptions ={
    connections?: number,
    gates?: number,
    selfconnections?: number,
    backconnections?: number,
}

export const Random: (
    input: number, 
    hidden: number, 
    output: number, 
    options: RandomNetworkOptions
) => INetwork = neataptic.architect.Random

export const Hopfield: (size: number) => INetwork = neataptic.architect.Random