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