()
| 228 | |
| 229 | |
| 230 | def main(): |
| 231 | Xtrain, Ytrain, Xtest, Ytest = getKaggleMNIST() |
| 232 | # dnn = DNN([1000, 750, 500]) |
| 233 | # dnn.fit(Xtrain, Ytrain, Xtest, Ytest, epochs=3) |
| 234 | # vs |
| 235 | dnn = DNN([1000, 750, 500]) |
| 236 | dnn.fit(Xtrain, Ytrain, Xtest, Ytest, pretrain=True, train_head_only=False, epochs=3) |
| 237 | # note: try training the head only too! what does that mean? |
| 238 | |
| 239 | |
| 240 | def test_single_autoencoder(): |
no test coverage detected