https://github.com/deepcs233/deeplearning-framework-for-recsys
https://github.com/deepcs233/deeplearning-framework-for-recsys
Last synced: 2 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/deepcs233/deeplearning-framework-for-recsys
- Owner: deepcs233
- Created: 2018-03-21T03:48:53.000Z (over 8 years ago)
- Default Branch: master
- Last Pushed: 2018-08-27T07:43:55.000Z (almost 8 years ago)
- Last Synced: 2025-03-11T01:46:44.083Z (over 1 year ago)
- Language: Python
- Size: 27.3 KB
- Stars: 2
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# DeepLearning-Framework-for-recsys
框架使用demo见demo.py
example
```python
# 初始化IO管理器
iM = ioManger(batch_size=16)
# 初始化网络
net = Net(iM)
# 添加全连接层
net.add(DotLayer(inputs=['bias_net'], outputs=['aa'], hiddenNum=1))
# 添加sigmoid激活函数
net.add(SigmoidLayer(inputs=['aa'], outputs=['Output']))
# 初始化SGD optimizer
sgd = SGD(lr=0.00003)#,momentum=0.99,nesterov=True)
# 初始化model
model = Model(net=net, optimizer=sgd, lossFunc=cross_entropy_error(), ioManger=iM)
# build model
model.build()
# 开始训练
model.fit(X=X, y=y, validation_split=0.12, epochs=1)
```