UNPKG

@modelx/model

Version:

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

25 lines (24 loc) 1.27 kB
import { TensorScriptOptions, TensorScriptProperties, Matrix, TensorScriptLayers } from './model_interface'; import { BaseNeuralNetwork } from './base_neural_network'; /** * Logistic Regression Classification with Tensorflow * @class LogisticRegression * @implements {BaseNeuralNetwork} */ export declare class LogisticRegression extends BaseNeuralNetwork { /** * @param {{layers:Array<Object>,compile:Object,fit:Object}} 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 classification 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 * @param {Array<Array<number>>} x_test - validation data independent variables * @param {Array<Array<number>>} y_test - validation data dependent variables */ generateLayers(x_matrix: Matrix, y_matrix: Matrix, layers: TensorScriptLayers, x_test: Matrix, y_test: Matrix): void; }