Permalink
Browse files

FunctionSet -> Chain, Functions -> Linksの変更(chainer1.5対応)を投入

  • Loading branch information...
ISP-Tetsuro-Kitajima committed Feb 4, 2016
1 parent 10a2b9c commit 83b6eacbc1028614a54dad990323d28580a50831
Showing with 6 additions and 5 deletions.
  1. +6 −5 tests/test_manager.py
View
@@ -3,8 +3,9 @@
import sys
import numpy as np
from chainer import FunctionSet, Variable, optimizers
from chainer import Chain, Variable, optimizers
import chainer.functions as F
import chainer.links as L
from sklearn.base import ClassifierMixin
from sklearn import cross_validation
@@ -14,10 +15,10 @@
class TestNNM(NNmanager, ClassifierMixin):
def __init__(self, logging=False):
model = FunctionSet(
l1=F.Linear(784, 100),
l2=F.Linear(100, 100),
l3=F.Linear(100, 10)
model = Chain(
l1=L.Linear(784, 100),
l2=L.Linear(100, 100),
l3=L.Linear(100, 10)
)
optimizer = optimizers.SGD()

0 comments on commit 83b6eac

Please sign in to comment.