Вход fit и evaluate должен иметь следующие признаки, иначе возникнет KeyError:
если weight_column_name не None: ключ=weight_column_name, значение=%%CODE_BLOCK_7%%
для столбца в feature_columns:
если isinstance(столбец, SparseColumn): ключ=имя_столбца, значение=%%CODE_BLOCK_10%%
если isinstance(столбец, WeightedSparseColumn): {ключ=имя_столбца, значение=%%CODE_BLOCK_12%%, ключ=имя_столбца_веса, значение=%%CODE_BLOCK_13%%%}
если isinstance(столбец, RealValuedColumn): ключ=имя_столбца, значение=%%CODE_BLOCK_15%%%
Аргументы
feature_columns
Итерируемый объект, содержащий все столбцы признаков, используемые моделью. Все элементы набора должны быть экземплярами классов, производными от FeatureColumn.
head
Экземпляр класса _Head.
model_dir
Каталог для сохранения параметров модели, графа и т. д. Это также можно использовать для загрузки контрольных точек из каталога в оценщик для продолжения обучения ранее сохраненной модели.
weight_column_name
Строка, определяющая имя столбца признака, представляющего веса. Используется для снижения или повышения веса примеров во время обучения. Будет умножаться на потерю примера.
optimizer
Экземпляр tf.Optimizer, используемый для обучения модели. Если None, будет использоваться оптимизатор Ftrl.
gradient_clip_norm
Значение float > 0. Если задано, градиенты ограничиваются их глобальной нормой с этим коэффициентом ограничения. См. tf.clip_by_global_norm для получения дополнительной информации.
_joint_weights
Если True, используйте один (возможно, раздробленный) переменный для хранения весов. Это быстрее, но требует, чтобы все столбцы признаков были разреженными и имели комбинировщик «сумма». Несовместимо с SDCAOptimizer.
config
Объект RunConfig, для настройки параметров выполнения.
feature_engineering_fn
Функция обработки признаков. Принимает признаки и метки, которые являются результатом input_fn, и возвращает признаки и метки, которые будут переданы в модель.
Возможные исключения
ValueError
если оптимизатор не поддерживается, например, SDCAOptimizer
Атрибуты
config
model_dir
Возвращает путь, в котором процесс оценки будет искать контрольные точки.
Экспортирует граф вывода в заданный каталог. (устарело)
Аргументы
export_dir
Строка, содержащая каталог для записи экспортированного графа и контрольных точек.
input_fn
Если use_deprecated_input_fn равно true, то функция, которая, учитывая Tensor строк Example, анализирует их в признаки, которые затем передаются модели. В противном случае, функция, которая не принимает аргументов и возвращает кортеж (признаки, метки), где признаки — это словарь со строковым ключом к Tensor, а метки — это Tensor, который в настоящее время не используется (и поэтому может быть None).
input_feature_key
Используется только если use_deprecated_input_fn равно false. Строковый ключ в словаре признаков, возвращенном input_fn, который соответствует исходным строкам ExampleTensor, которые будут приняты экспортированной моделью в качестве входных данных. Может быть только None, если вы используете пользовательскую signature_fn, которая не использует первый аргумент (примеры).
use_deprecated_input_fn
Определяет формат подписи input_fn.
signature_fn
Функция, которая возвращает стандартную подпись и карту имен подписей, учитывая Tensor строк Example, dictTensor для признаков и Tensor или dictTensor для предсказаний.
prediction_key
Ключ для тензора в словаре predictions (выход из model_fn) для использования в качестве входного значения predictions в signature_fn. Необязательно. Если None, предсказания будут переданы signature_fn без фильтрации.
default_batch_size
Значение по умолчанию размера пакета Example плейсхолдера.
exports_to_keep
Количество экспортов для хранения.
checkpoint_path
Путь к контрольной точке модели, которая должна быть экспортирована. Если это None (по умолчанию), будет использоваться последняя контрольная точка в export_dir.
Возвращаемое значение
Путь к строке экспортированного каталога. Примечание: эта функция была добавлена ок. 2016/09/25; клиентам, зависящим от возвращаемого значения, может потребоваться обработать случай, когда эта функция возвращает None, потому что подклассы не возвращают значение.
Экспортирует граф вывода как SavedModel в заданный каталог.
Аргументы
export_dir_base
Строка, содержащая каталог для записи экспортированного графа и контрольных точек.
serving_input_fn
Функция, которая не принимает аргументов и возвращает InputFnOps.
default_output_alternative_key
Имя заголовка для обработки, когда не указан. Не нужно для моделей с единственным заголовком.
assets_extra
Словарь, определяющий способ заполнения каталога assets.extra в экспортированном SavedModel. Каждый ключ должен указывать путь назначения (включая имя файла) относительно каталога assets.extra. Соответствующее значение — это полный путь исходного файла, который нужно скопировать. Например, простой случай копирования единственного файла без переименования задается как {'my_asset_file.txt': '/path/to/my_asset_file.txt'}.
as_text
Флаг записи протокола SavedModel в текстовом формате.
checkpoint_path
Путь к контрольной точке для экспорта. Если None (по умолчанию), выбирается самая последняя контрольная точка, найденная в каталоге модели.
graph_rewrite_specs
Итерируемый объект GraphRewriteSpec. Каждый элемент создаст отдельную MetaGraphDef в экспортированном SavedModel, помеченный и переписанный, как указано. По умолчанию — один элемент с тегом «serve» и без переписывания.
Инкрементная подгонка на наборе образцов. (устаревшие аргументы)
Ожидается, что этот метод вызывается несколько раз последовательно на различных или одинаковых фрагментах набора данных. Это может реализовать итеративное обучение или обучение вне памяти/онлайн.
Это особенно полезно, когда весь набор данных слишком велик, чтобы поместиться в памяти одновременно. Или когда модели требуется много времени для схождения, и вы хотите разделить обучение на подчасти.
Аргументы
x
Матрица формы [n_samples, n_features...]. Может быть итератором, который возвращает массивы признаков. Образцы входных данных для обучения модели. Если задано, input_fn должно быть None.
y
Вектор или матрица [n_samples] или [n_samples, n_outputs]. Может быть итератором, который возвращает массив меток. Значения меток обучения (метки классов в классификации, действительные числа в регрессии). Если задано, input_fn должно быть None.
input_fn
Функция входных данных. Если задано, x, y, и batch_size должны быть None.
steps
Количество шагов, на которых нужно обучить модель. Если None, обучаться вечно.
batch_size
размер мини-пакета для использования с входными данными, по умолчанию равняется первой размерности x. Должно быть None если input_fn предоставлено.
monitors
Список экземпляров подкласса BaseMonitor. Используется для обратных вызовов внутри цикла обучения.
Возвращает
self, для цепочки.
Исключения
ValueError
Если хотя бы один из x и y предоставлен, и input_fn предоставлен.
Возвращает прогнозы для заданных признаков. (устаревшие аргументы)
Аргументы
x
Матрица формы [n_samples, n_features...]. Может быть итератором, который возвращает массивы признаков. Образцы входных данных для обучения модели. Если задано, input_fn должно быть None.
input_fn
Функция входных данных. Если задано, x и 'batch_size' должны быть None.
batch_size
Переопределение размера пакета по умолчанию. Если задано, 'input_fn' должно быть 'None'.
outputs
список str, имя выходного значения для прогноза. Если None, возвращает все.
as_iterable
Если True, возвращает итерируемый объект, который продолжает выдавать прогнозы для каждого примера до истощения входных данных. Примечание: входные данные должны завершаться, если вы хотите, чтобы итерируемый объект завершился (например, убедитесь, что вы передали num_epochs=1, если вы используете что-то вроде read_batch_features).
iterate_batches
Если True, выдает весь пакет сразу, вместо разбиения пакета на отдельные образцы. Актуально только тогда, когда as_iterable True.
Возвращает
Массив NumPy предсказанных классов или регрессионных значений, если конструктор model_fn возвращает Tensor для predictions или массив NumPy, если dict возвращает dict. Возвращает итерируемый объект прогнозов, если as_iterable True.
Метод работает как с простыми оценщиками, так и с вложенными объектами (такими как конвейеры). Первые имеют параметры вида <component>__<parameter>, чтобы можно было обновить каждый компонент вложенного объекта.