DBN for Univariate series with a dicision label

albertbup / deep-belief-network

A Python implementation of Deep Belief Networks built upon NumPy and TensorFlow with scikit-learn compatibility

MIT License

481 stars 212 forks source link

DBN for Univariate series with a dicision label #30

Closed PradyumnaSripad closed 6 years ago

PradyumnaSripad commented 6 years ago

Greetings sir,

My code for univariate classification for 5 classes is

X=df['value'].values;y = df['class'] dbn = UnsupervisedDBN(hidden_layers_structure=[1,5], batch_size=10, learning_rate_rbm=0.06, n_epochs_rbm=20, activation_function='sigmoid')

The error i got is Please help me to rectify the same! issue

albertbup commented 6 years ago

Hi PradyumnaSripad,

I've solved the bug in PR #31 Re-install the package and try again, it should work now.

Best.