Spec-Zone.ru › TensorFlow 1.15

tf.contrib.learn.LogisticRegressor

Создаёт оценщик логистической регрессии для бинарной классификации.

tf.contrib.learn.LogisticRegressor(
    model_fn, thresholds=None, model_dir=None, config=None,
    feature_engineering_fn=None
)

ЭТОТ КЛАСС УСТЕРЕЛ. Смотрите contrib/learn/README.md для общих инструкций по миграции.

Этот метод предоставляет базовый оценщик с некоторыми дополнительными метриками для пользовательских моделей бинарной классификации, включая AUC, точность/полноту и точность.

Пример:

# See tf.contrib.learn.Estimator(...) for details on model_fn structure
def my_model_fn(...):
  pass

estimator = LogisticRegressor(model_fn=my_model_fn)

# Input builders
def input_fn_train:
  pass

estimator.fit(input_fn=input_fn_train)
estimator.predict(x=x)
Аргументы
model_fn Функция модели с сигнатурой: (features, labels, mode) -> (predictions, loss, train_op). Ожидается, что возвращаемые предсказания будут вероятностями в диапазоне [0.0, 1.0].
thresholds Список точек с плавающей запятой, используемых для метрик точности, полноты и точности. Если None, по умолчанию [0.5].
model_dir Каталог для сохранения параметров модели, графиков и т. д. Его также можно использовать для загрузки контрольных точек из каталога в оценщик для продолжения обучения ранее сохранённой модели.
config Объект конфигурации RunConfig.
feature_engineering_fn Функция обработки признаков. Принимает признаки и метки, которые являются результатом input_fn и возвращает признаки и метки, которые будут поданы в модель.
Возвращает
Экземпляр Estimator .

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/learn/LogisticRegressor

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API