6

scikit で RBM を使用したいと思います。他の多くの分類器と同様に、RBM を定義してトレーニングできます。

from sklearn.neural_network import BernoulliRBM
clf = BernoulliRBM(random_state=0, verbose=True)
clf.fit(X_train, y_train)

しかし、私を予測させる関数が見つからないようです。scikit で次のいずれかに相当するものを探しています。

y_score = clf.decision_function(X_test)
y_score = clf.predict(X_test)

どちらの関数も BernoulliRBM には存在しません。

4

1 に答える 1