UNPKG

@stdlib/ml

Version:

Machine learning algorithms.

114 lines (104 loc) 4.34 kB
/** * @license Apache-2.0 * * Copyright (c) 2018 The Stdlib Authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ 'use strict'; // MODULES // var isNonNegativeNumber = require( '@stdlib/assert/is-nonnegative-number' ).isPrimitive; var isPositiveNumber = require( '@stdlib/assert/is-positive-number' ).isPrimitive; var isNumber = require( '@stdlib/assert/is-number' ).isPrimitive; var isBoolean = require( '@stdlib/assert/is-boolean' ).isPrimitive; var isArrayLikeObject = require( '@stdlib/assert/is-array-like-object' ); var isObject = require( '@stdlib/assert/is-plain-object' ); var hasOwnProp = require( '@stdlib/assert/has-own-property' ); var contains = require( '@stdlib/assert/contains' ); var format = require( '@stdlib/string/format' ); var LEARNING_RATES = require( './learning_rates.json' ); var LOSS_FUNCTIONS = require( './loss_functions.json' ); // MAIN // /** * Validates function options. * * @private * @param {Object} opts - destination object * @param {Options} options - function options * @param {PositiveNumber} [options.lambda] - regularization parameter * @param {ArrayLikeObject} [options.learningRate] - learning rate function * @param {string} [options.loss] - loss function * @param {boolean} [options.intercept] - specifies whether an intercept should be included * @returns {(Error|null)} null or an error object * * @example * var opts = {}; * var options = {}; * var err = validate( opts, options ); * if ( err ) { * throw err; * } */ function validate( opts, options ) { var name; if ( !isObject( options ) ) { return new TypeError( format( 'invalid argument. Options argument must be an object. Value: `%s`.', options ) ); } if ( hasOwnProp( options, 'intercept' ) ) { opts.intercept = options.intercept; if ( !isBoolean( opts.intercept ) ) { return new TypeError( format( 'invalid option. `%s` option must be a boolean. Option: `%s`.', 'intercept', opts.intercept ) ); } } if ( hasOwnProp( options, 'lambda' ) ) { opts.lambda = options.lambda; if ( !isNonNegativeNumber( opts.lambda ) ) { return new TypeError( format( 'invalid option. `%s` option must be a nonnegative number. Option: `%s`.', 'lambda', opts.lambda ) ); } } if ( hasOwnProp( options, 'learningRate' ) ) { if ( !isArrayLikeObject( options.learningRate ) ) { return new TypeError( format( 'invalid option. `%s` option must be an array-like object. Option: `%s`.', 'learningRate', options.learningRate ) ); } name = options.learningRate[ 0 ]; opts.learningRate[ 0 ] = name; if ( !contains( LEARNING_RATES, name ) ) { return new TypeError( format( 'invalid option. First `%s` option must be one of the following: "%s". Option: `%s`.', 'learningRate', LEARNING_RATES.join( '", "' ), name ) ); } if ( options.learningRate.length > 1 ) { if ( name === 'constant' || name === 'invscaling' ) { opts.learningRate[ 1 ] = options.learningRate[ 1 ]; if ( !isPositiveNumber( opts.learningRate[ 1 ] ) ) { return new TypeError( format( 'invalid option. Second `%s` option must be a positive number. Option: `%s`.', 'learningRate', opts.learningRate[ 1 ] ) ); } } } if ( options.learningRate.length > 2 ) { if ( name === 'invscaling' ) { opts.learningRate[ 2 ] = options.learningRate[ 2 ]; if ( !isNumber( opts.learningRate[ 2 ] ) ) { return new TypeError( format( 'invalid option. Third `%s` option must be a number. Option: `%s`.', 'learningRate', opts.learningRate[ 2 ] ) ); } } } } if ( hasOwnProp( options, 'loss' ) ) { opts.loss = options.loss; if ( !contains( LOSS_FUNCTIONS, opts.loss ) ) { return new TypeError( format( 'invalid option. `%s` option must be one of the following: "%s". Option: `%s`.', 'loss', LOSS_FUNCTIONS.join( '", "' ), opts.loss ) ); } } return null; } // EXPORTS // module.exports = validate;