01-深度学习框架

  • tensorflow
  • keras
  • pytorch
  • Caffe
  • PaddlePaddle
  • ...

keras优势

  • python 的机器学习框架,专注于机器学习的过程， 并非语法本身
  • 容易上手，方便理解
  • 底层支持，tensorflow，cntk， theano等框架

keras常见API介绍

导入keras包，创建一个线性的模型

1
2
3
from keras.models import Sequential

model = Sequential()

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
from keras.layers.core import Dense, Dropout, Flatten
from keras.layers import Conv2D, MaxPooling2D

model.add(Conv2D())
model.add(MaxPooling2D())
model.add(Flatten())
model.add(Dense())
model.add(Dense())

model.compile()

model.fit()

model.evaluate()

model.predict()