2016-07-29 96 views
-2

這裏我開發了一個神經網絡分類器來解決泰坦尼克問題。python中的神經網絡分類器

from sknn.mlp import Classifier, Layer 

nn = Classifier(
    layers=[ 
     Layer("Maxout", units=100, pieces=2), 
     Layer("Softmax")], 
    learning_rate=0.001, 
    n_iter=25) 


nn.fit(X_train, y_train) 

我得到這個錯誤,我已經嘗試了很多解決它,但沒有與我的作品。 請幫我

TypeError: init() got an unexpected keyword argument 'pieces'

+0

這本來是很容易通過查看層)的可能參數(解決 – Andy

回答

0

Layer簽名不定義任何參數調用pieces。要創建兩層相同的參數,你必須定義兩次Layer對象:

layers=[ 
    Layer("Sigmoid", units=100), 
    Layer("Sigmoid", units=100), 
    Layer("Softmax", units=1)] # The units parameter is not optional 

更何況,"Maxout"看起來並不像一個Layer類型。不知道你在哪裏找到。

Specifically, options are Rectifier , Sigmoid , Tanh , and ExpLin for non-linear layers and Linear or Softmax for output layers

+0

我在這裏找到https://scikit-neuralnetwork.readthedocs.io/en/latest/guide_model.html#classification – Haifa

+0

我已嘗試過您的建議,但出現錯誤,AssertionError:數據集大小與輸出層中的單位不匹配。 – Haifa