AI如何快速准确地学习乘法?( 二 )


在配备16 GB RAM的NVidia 940 MX 4 GB GPU上,训练过程大约需要20秒才能完成100次迭代 。确实,这是快速的学习 。
结果最后,该测试我们的模型了 。让我们在test_array中获取任何五个数字,并使用上面构建的模型来获取预测 。理想情况下,输出应为test_array元素乘以5 。
test_array=np.array([4,27,100,121,9])
print(model.predict([test_array]))
做出最终预测的时间 。
[[19.9995174407959], [134.9995880126953], [499.9997253417969], [604.9998168945312], [44.99952697753906]]
我们看到我们的模型几乎就在那里 。实际值为[20,135,500,605,45],如果将预测四舍五入到小数点后两位,我们将看到预测与预期结果相同 。

AI如何快速准确地学习乘法?

文章插图
> Canva.com
 
这样就可以了,一个简单的AI现在已经学习了乘法 。但是,要考虑:尝试给定负值并检查输出,您会得到什么? 我想让您尝试,试验并提出使算法更准确的技术 。另外,尝试更改体系结构,例如添加更多的层和神经元,并查看指标的差异 。如果您有任何疑问,请在评论中让我知道 。
我希望您喜欢这篇文章,并幻想构建一个AI,该AI被编程为执行非常简单的乘法任务 。下次见 。
(本文翻译自Nemath Ahmed的文章《AI Learns to Multiply in 20 Seconds》,参考:
https://towardsdatascience.com/ai-learns-to-multiply-ce844c68aefe)

【AI如何快速准确地学习乘法?】


推荐阅读