1. tflearn.data_utils

  • from tflearn.data_utils import to_categorical
    • one_hot 编码;
    • 第一个参数为属性列,第二个参数接受类别个数;

2. tflearn.layers.core

  • from tflearn.layers.core import
    • input_data,
    • fully_connected,
    • dropout

3. tflearn.layers.conv

  • from tflearn.layers.conv import
    • conv_2d,
    • max_pool_2d
    • global_avg_pool

4. tflearn.layers.estimator

  • from tflearn.layers.estimator import
    • regression

5. model

  • tlearn.DNN()