Deepy-基於 Numpy 的小巧優雅簡單的深度學習庫

只是為了好玩，就用python編寫了一個深度學習庫。

它使用numpy進行計算。 API類似於PyTorch的API。

Github項目地址：https://github.com/kaszperro/deepy

示例

在示例目錄中有一個線性分類器，其準確率超過96%。

順序模型的創建：

損失：

fromdeepy.moduleimportLinear

fromdeepy.autograd.lossesimportCrossEntropyLoss, MSELoss

fromdeepy.variableimportVariable

importnumpyasnp

my_model = Linear(10,10)

loss1 = CrossEntropyLoss()

loss2 = MSELoss()

good_output = Variable(np.zeros((10,10)))

model_input = Variable(np.ones((10,10)))

model_output = my_model(model_input)

error = loss1(good_output, model_output)# now you can propagate error backwards:error.backward()

優化：

fromdeepy.moduleimportLinear

fromdeepy.autograd.lossesimportCrossEntropyLoss, MSELoss

fromdeepy.variableimportVariable

fromdeepy.autograd.optimizersimportSGD

importnumpyasnp

my_model = Linear(10,10)

loss1 = CrossEntropyLoss()

loss2 = MSELoss()

optimizer1 = SGD(my_model.get_variables_list())

good_output = Variable(np.zeros((10,10)))

model_input = Variable(np.ones((10,10)))

model_output = my_model(model_input)

error = loss1(good_output, model_output)

# now you can propagate error backwards:

error.backward()

# and then optimizer can update variables:

optimizer1.zero_grad()

optimizer1.step()

項目地址：https://github.com/kaszperro/deepy

【AI求職百題斬 - 每日一題】

趕緊來看看今天的題目吧！

想知道正確答案？

點擊公眾號菜單欄【每日一題】【每日一題】或在公眾號回復「0129」即可答題獲取！

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！