UNPKG

@modelx/model

Version:

Deep Learning Classification, LSTM Time Series, Regression and Multi-Layered Perceptrons with Tensorflow

23 lines (22 loc) 1.09 kB
import { TensorScriptOptions, TensorScriptProperties, Matrix, TensorScriptLayers } from './model_interface'; import { BaseNeuralNetwork } from './base_neural_network'; /** * Deep Learning Regression with Tensorflow * @class DeepLearningRegression * @implements {BaseNeuralNetwork} */ export declare class DeepLearningRegression extends BaseNeuralNetwork { /** * @param {{layers:Array<Object>,compile:Object,fit:Object,layerPreference:String}} options - neural network configuration and tensorflow model hyperparameters * @param {{model:Object,tf:Object,}} properties - extra instance properties */ constructor(options?: TensorScriptOptions, properties?: TensorScriptProperties); /** * Adds dense layers to tensorflow regression model * @override * @param {Array<Array<number>>} x_matrix - independent variables * @param {Array<Array<number>>} y_matrix - dependent variables * @param {Array<Object>} layers - model dense layer parameters */ generateLayers(x_matrix: Matrix, y_matrix: Matrix, layers: TensorScriptLayers): void; }