INNER CODE UNIT · Python

learning_rate

deepchem/deepchem · contrib/DiabeticRetinopathy/run.py:22

    learning_rate=1e-5,
    augment=True,
    model_dir='./test_model')
if not os.path.exists('./test_model'):
  os.mkdir('test_model')
model.build()
#model.restore()
metrics = [
    dc.metrics.Metric(DRAccuracy, mode='classification'),
    dc.metrics.Metric(QuadWeightedKappa, mode='classification')
]
cm = [dc.metrics.Metric(ConfusionMatrix, mode='classification')]

logger = logging.getLogger('deepchem.models.tensorgraph.tensor_graph')
logger.setLevel(logging.DEBUG)
for i in range(10):
  model.fit(train, nb_epoch=10, checkpoint_interval=3512)
  model.evaluate(train, metrics)

View source record →