Home
API
Examples
Github
Home
API
Examples
Github
  • API

    • cluster

      • KMeans
    • datasets

      • Boston
      • HeartDisease
      • Iris
    • decomposition

      • PCA
    • ensemble

      • BaggingClassifier
      • RandomForestClassifier
    • feature_extraction

      • CountVectorizer
    • linear_model

      • Lasso
      • LinearRegression
      • LogisticRegression
      • Ridge
      • SGDClassifier
      • SGDRegressor
    • metrics

      • accuracyScore
      • confusion_matrix
      • mean_absolute_error
      • mean_squared_error
      • mean_squared_log_error
      • zeroOneLoss
    • model_selection

      • KFold
      • train_test_split
    • naive_bayes

      • GaussianNB
      • MultinomialNB
    • neighbors

      • KNeighborsClassifier
    • preprocessing

      • Binarizer
      • Imputer
      • LabelEncoder
      • MinMaxScaler
      • OneHotEncoder
      • PolynomialFeatures
      • add_dummy_feature
      • normalize
    • svm

      • BaseSVM
      • NuSVC
      • NuSVR
      • OneClassSVM
      • SVC
      • SVR
    • tree

      • DecisionTreeClassifier

metrics.mean_squared_error

▸ mean_squared_error(y_true: `object`, y_pred: `object`, __namedParameters: `object`)

Usage

import { mean_squared_error } from 'machinelearn/metrics';

const y_true = [3, -0.5, 2, 7];
const y_pred = [2.5, 0.0, 2, 8];

console.log(mean_squared_error(y_true, y_pred));
// result: 0.375

const y_true1 = [[0.5, 1], [-1, 1], [7, -6]];
const y_pred1 = [[0, 2], [-1, 2], [8, -5]];

console.log(mean_squared_error(y_true1, y_pred1));
// result: 0.7083333134651184

Defined in metrics/regression.ts:94

Parameters:

ParamTypeDefaultDescription
y_truenumber[] or number[][]nullGround truth (correct) target values.
y_prednumber[] or number[][]nullEstimated target values.
options.sample_weightnumbernull

Returns:

number

Prev
mean_absolute_error
Next
mean_squared_log_error