皮托克用的像keras的包装纸
keraTorch的Python项目详细描述
角膜弓
Implementing Keras clone with pytorch backend.
安装
pip install keratorch
如何使用
fromkeraTorch.modelimportSequentialfromkeraTorch.layersimport*fromkeraTorch.lossesimport*
数据:
^{pr2}$((50000, 784), (50000,), (10000, 784), (10000,))
模型定义:
model=Sequential()model.add(Dense(100,x_train.shape[1],activation='relu'))model.add(Dense(50,activation='relu'))model.add(Dense(10))model.add(Activation('softmax'))
实际上没有编译任何东西,只是看起来像keras,我们指定如下的损失。ce4softmax
表示softmax损失的交叉熵。在
model.compile(ce4softmax)
为Fastai的学习率查找器挖掘最佳学习率:
bs=256model.lr_find(x_train,y_train,bs=bs)
Min numerical gradient: 9.12E-03
Min loss divided by 10: 1.45E-02
我们有相同的.fit
和.predict
函数:
model.fit(x_train,y_train,bs,epochs=10,lr=1e-2)
preds=model.predict(x_valid)accuracy=(preds.argmax(axis=-1)==y_valid).mean()print(f'Predicted accuracy is {accuracy:.2f}')
Predicted accuracy is 0.81
- 项目
标签: