Model fitting

We have previously said that RBM is often used as a feature in a wide variety of classification problems. It's time to see how to do it. The first thing to do is to use the BernoulliRBM function of the sklearn.neural_network module.

sklearn is a free machine learning library for the Python programming language. It features various classification, regression, and clustering algorithms, including support vector machines, random forests, gradient boosting, k-means, and DBSCAN. And it is designed to interoperate with the Python numerical and scientific libraries NumPy and SciPy.

In the sklearn library, the sklearn.neural_network module includes models based on neural networks. In this module, the BernoulliRBM function fits a Bernoulli ...

Get Hands-On Machine Learning on Google Cloud Platform now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.