tf.contrib.learn的简介
tf.contrib.learn 是 TensorFlow 提供的一个机器学习高级 API 模块,让用户可以更方便的配置、训练和评估各种各样的机器学习模型,里面内置了很多模型可以直接调用,其实,就是类似最经典的机器学习库 scikit-learn 的 API 。很多调用方法都很相同。
1、常用的模型
线性分类 tf.contrib.learn.LinearClassifier
线性回归 tf.contrib.learn.LinearRegressor
逻辑斯谛回归 tf.contrib.learn.LogisticRegressor
K-means 聚类 tf.contrib.learn.KMeansClustering
神经网络分类器 tf.contrib.learn.DNNClassifier
神经网络回归器 tf.contrib.learn.DNNRegressor
广度深度回归 tf.contrib.learn.DNNLinearCombinedRegressor
广度深度分类 tf.contrib.learn.DNNLinearCombinedClassifier
注意:该模块已被弃用,请仔细查看API说明,https://github.com/tensorflow/tensorflow/blob/r1.13/tensorflow/contrib/learn/README.md
Many canned estimators (subclasses of Estimator) have equivalents in core exposed under tf.estimator: DNNClassifier, DNNRegressor, DNNEstimator, LinearClassifier, LinearRegressor, LinearEstimator, DNNLinearCombinedClassifier,DNNLinearCombinedRegressor and DNNLinearCombinedEstimator.
To migrate to the new api, users need to take the following steps:
Replace tf.contrib.learn with tf.estimator.
tf.contrib.learn的使用方法
Modules
datasets module: Dataset utilities and synthetic/reference datasets (deprecated).
graph_actions module: High level operations on graphs (deprecated).
head module: Abstractions for the head(s) of a model (deprecated).
io module: Tools to allow different io formats (deprecated).
learn_runner module: Utilities to run and tune an Experiment (deprecated).
models module: Various high level TF models (deprecated).
monitors module: Monitors instrument the training process (deprecated).
ops module: Various TensorFlow Ops (deprecated).
preprocessing module: Preprocessing tools useful for building models (deprecated).
utils module: TensorFlow Learn Utils (deprecated).
Classes
class BaseEstimator: Abstract BaseEstimator class to train and evaluate TensorFlow models.
class DNNClassifier: A classifier for TensorFlow DNN models.
class DNNEstimator: A Estimator for TensorFlow DNN models with user specified _Head.
class DNNLinearCombinedClassifier: A classifier for TensorFlow Linear and DNN joined training models.
class DNNLinearCombinedEstimator: An estimator for TensorFlow Linear and DNN joined training models.
class DNNLinearCombinedRegressor: A regressor for TensorFlow Linear and DNN joined training models.
class DNNRegressor: A regressor for TensorFlow DNN models.
class DynamicRnnEstimator: Dynamically unrolled RNN (deprecated).
class Estimator: Estimator class is the basic TensorFlow model trainer/evaluator.
class Evaluable: Interface for objects that are evaluatable by, e.g., Experiment.
class Experiment: Experiment is a class containing all information needed to train a model.
class ExportStrategy: A class representing a type of model export.
class Head: Interface for the head/top of a model.
class InputFnOps: A return type for an input_fn (deprecated).
class KMeansClustering: An Estimator for K-Means clustering.
class LinearClassifier: Linear classifier model.
class LinearEstimator: Linear model with user specified head.
class LinearRegressor: Linear regressor model.
class MetricSpec: MetricSpec connects a model to metric functions.
class ModeKeys: Standard names for model modes (deprecated).
class ModelFnOps: Ops returned from a model_fn.
class NanLossDuringTrainingError: Unspecified run-time error.
class NotFittedError: Exception class to raise if estimator is used before fitting.
class PredictionKey: THIS CLASS IS DEPRECATED.
class ProblemType: Enum-like values for the type of problem that the model solves.
class RunConfig: This class specifies the configurations for an Estimator run.
class SKCompat: Scikit learn wrapper for TensorFlow Learn Estimator.
class SVM: Support Vector Machine (SVM) model for binary classification.
class TaskType: DEPRECATED CLASS.
class Trainable: Interface for objects that are trainable by, e.g., Experiment.
Functions
LogisticRegressor(...): Builds a logistic regression Estimator for binary classification.
binary_svm_head(...): Creates a Head for binary classification with SVMs. (deprecated)
build_parsing_serving_input_fn(...): Build an input_fn appropriate for serving, expecting fed tf.Examples. (deprecated)
evaluate(...): Evaluate a model loaded from a checkpoint. (deprecated)
extract_dask_data(...): Extract data from dask.Series or dask.DataFrame for predictors. (deprecated)
extract_dask_labels(...): Extract data from dask.Series or dask.DataFrame for labels. (deprecated)
extract_pandas_data(...): Extract data from pandas.DataFrame for predictors. (deprecated)
extract_pandas_labels(...): Extract data from pandas.DataFrame for labels. (deprecated)
extract_pandas_matrix(...): Extracts numpy matrix from pandas DataFrame. (deprecated)
infer(...): Restore graph from restore_checkpoint_path and run output_dict tensors. (deprecated)
infer_real_valued_columns_from_input(...): Creates FeatureColumn objects for inputs defined by input x. (deprecated)
infer_real_valued_columns_from_input_fn(...): Creates FeatureColumn objects for inputs defined by input_fn. (deprecated)
make_export_strategy(...): Create an ExportStrategy for use with Experiment. (deprecated)
multi_class_head(...): Creates a Head for multi class single label classification. (deprecated)
multi_head(...): Creates a MultiHead stemming from same logits/hidden layer. (deprecated)
multi_label_head(...): Creates a Head for multi label classification. (deprecated)
no_op_train_fn(...): DEPRECATED FUNCTION
poisson_regression_head(...): Creates a Head for poisson regression. (deprecated)
read_batch_examples(...): Adds operations to read, queue, batch Example protos. (deprecated)
read_batch_features(...): Adds operations to read, queue, batch and parse Example protos. (deprecated)
read_batch_record_features(...): Reads TFRecord, queues, batches and parses Example proto. (deprecated)
read_keyed_batch_examples(...): Adds operations to read, queue, batch Example protos. (deprecated)
read_keyed_batch_examples_shared_queue(...): Adds operations to read, queue, batch Example protos. (deprecated)
read_keyed_batch_features(...): Adds operations to read, queue, batch and parse Example protos. (deprecated)
read_keyed_batch_features_shared_queue(...): Adds operations to read, queue, batch and parse Exampleprotos.(deprecated)
regression_head(...): Creates a Head for linear regression. (deprecated)
run_feeds(...): See run_feeds_iter(). Returns a list instead of an iterator. (deprecated)
run_n(...): Run output_dict tensors n times, with the same feed_dict each run. (deprecated)
train(...): Train a model. (deprecated)
tf.contrib.learn的案例应用
更新……