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()
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:深度学习框架 —— tflearn 的学习 - Python技术站