Перейти к основному содержимому
Перейти к основному содержимому

stochasticLogisticRegression

Эта функция реализует стохастическую логистическую регрессию. Она может быть использована для решения задачи бинарной классификации, поддерживает те же настраиваемые параметры, что и stochasticLinearRegression, и работает аналогичным образом.

Parameters

Параметры точно такие же, как и в stochasticLinearRegression: learning rate, l2 regularization coefficient, mini-batch size, method for updating weights. Для получения дополнительной информации смотрите parameters.

1. Обучение

См. раздел Fitting в описании stochasticLinearRegression.

Предсказанные метки должны находиться в [-1, 1].

2. Предсказание

Используя сохраненное состояние, мы можем предсказать вероятность того, что объект имеет метку 1.

Запрос вернет столбец вероятностей. Обратите внимание, что первым аргументом evalMLMethod является объект AggregateFunctionState, следующими являются столбцы признаков.

Мы также можем установить границу вероятности, что позволяет назначать элементам различные метки.

Тогда результатом будут метки.

test_data — это таблица, аналогичная train_data, но может не содержать целевого значения.

Смотрите также