Home | History | Annotate | Download | only in learn 
      1 # Copyright 2016 The TensorFlow Authors. All Rights Reserved.
      2 #
      3 # Licensed under the Apache License, Version 2.0 (the "License");
      4 # you may not use this file except in compliance with the License.
      5 # You may obtain a copy of the License at
      6 #
      7 #     http://www.apache.org/licenses/LICENSE-2.0
      8 #
      9 # Unless required by applicable law or agreed to in writing, software
     10 # distributed under the License is distributed on an "AS IS" BASIS,
     11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     12 # See the License for the specific language governing permissions and
     13 # limitations under the License.
     14 # ==============================================================================
     15 
     16 # TODO(ptucker,ipolosukhin): Improve descriptions.
     17 """High level API for learning.
     18 
     19 See the @{$python/contrib.learn} guide.
     20 
     21 @@BaseEstimator
     22 @@Estimator
     23 @@Trainable
     24 @@Evaluable
     25 @@KMeansClustering
     26 @@ModeKeys
     27 @@ModelFnOps
     28 @@MetricSpec
     29 @@PredictionKey
     30 @@DNNClassifier
     31 @@DNNEstimator
     32 @@DNNRegressor
     33 @@DNNLinearCombinedRegressor
     34 @@DNNLinearCombinedEstimator
     35 @@DNNLinearCombinedClassifier
     36 @@DynamicRnnEstimator
     37 @@LinearClassifier
     38 @@LinearEstimator
     39 @@LinearRegressor
     40 @@LogisticRegressor
     41 @@StateSavingRnnEstimator
     42 @@SVM
     43 @@SKCompat
     44 
     45 @@Head
     46 @@multi_class_head
     47 @@multi_label_head
     48 @@binary_svm_head
     49 @@regression_head
     50 @@poisson_regression_head
     51 @@multi_head
     52 @@no_op_train_fn
     53 
     54 @@Experiment
     55 @@ExportStrategy
     56 @@TaskType
     57 
     58 @@NanLossDuringTrainingError
     59 @@RunConfig
     60 @@evaluate
     61 @@infer
     62 @@run_feeds
     63 @@run_n
     64 @@train
     65 
     66 @@extract_dask_data
     67 @@extract_dask_labels
     68 @@extract_pandas_data
     69 @@extract_pandas_labels
     70 @@extract_pandas_matrix
     71 @@infer_real_valued_columns_from_input
     72 @@infer_real_valued_columns_from_input_fn
     73 @@read_batch_examples
     74 @@read_batch_features
     75 @@read_batch_record_features
     76 @@read_keyed_batch_examples
     77 @@read_keyed_batch_examples_shared_queue
     78 @@read_keyed_batch_features
     79 @@read_keyed_batch_features_shared_queue
     80 
     81 @@InputFnOps
     82 @@ProblemType
     83 @@build_parsing_serving_input_fn
     84 @@make_export_strategy
     85 """
     86 
     87 from __future__ import absolute_import
     88 from __future__ import division
     89 from __future__ import print_function
     90 
     91 # pylint: disable=wildcard-import
     92 from tensorflow.contrib.learn.python.learn import *
     93 # pylint: enable=wildcard-import
     94 
     95 from tensorflow.contrib.learn.python.learn import learn_runner_lib as learn_runner
     96 
     97 from tensorflow.python.util.all_util import remove_undocumented
     98 
     99 _allowed_symbols = ['datasets', 'head', 'io', 'learn_runner', 'models',
    100                     'monitors', 'NotFittedError', 'ops', 'preprocessing',
    101                     'utils', 'graph_actions']
    102 
    103 remove_undocumented(__name__, _allowed_symbols)
    104