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

stochasticLogisticRegression

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

Параметры

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

1. Обучение

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

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

2. Прогнозирование

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

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

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

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

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

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