Python Trainer.ModelProcessor Examples

Example #1

File: test.py Project: BaiDingHub/neuralNet

# In[ ]:

x = X_train[:3000]
y = Y_train[:3000]

# In[ ]:

#loss,grads = model.loss(x,y)

# In[ ]:

#print(loss)

# In[ ]:

processor = Trainer.ModelProcessor(model, x, y)  #训练器

# In[ ]:

loss_history, acc_history = processor.train(epoch=10,
                                            iterations=50,
                                            printFreq=20)

# In[ ]:

test_x = X_test[:100]
test_y = Y_test[:100]

# In[ ]:

score, acc = model.predict(test_x, test_y)