In a context of machine learning, numerous tasks can be seen as prediction problem. For example, this user guide provides solutions for:
For any kinds of prediction problems, we generally provide a set of input-output pairs as:
["1:0.001","4:0.23","35:0.0035",...]
Once a prediction model has been constructed based on the samples, the model can make prediction for unforeseen inputs.
In order to train prediction models, an algorithm so-called stochastic gradient descent (SGD) is normally applied. You can learn more about this from the following external resources:
Importantly, depending on types of output value, prediction problem can be categorized into regression and classification problem.
The goal of regression is to predict real values as shown below:
features (input) | target real value (output) |
---|---|
[“1:0.001”,“4:0.23”,“35:0.0035”,...] | 21.3 |
[“1:0.2”,“3:0.1”,“13:0.005”,...] | 6.2 |
[“5:1.3”,“22:0.0.089”,“77:0.0001”,...] | 17.1 |
... | ... |
In practice, target values could be any of small/large float/int negative/positive values. Our CTR prediction tutorial solves regression problem with small floating point target values in a 0-1 range, for example.
While there are several ways to realize regression by using Hivemall, train_regressor()
is one of the most flexible functions. This feature is explained in this page.
In contrast to regression, output for classification problems should be (integer) labels:
features (input) | label (output) |
---|---|
[“1:0.001”,“4:0.23”,“35:0.0035”,...] | 0 |
[“1:0.2”,“3:0.1”,“13:0.005”,...] | 1 |
[“5:1.3”,“22:0.0.089”,“77:0.0001”,...] | 1 |
... | ... |
In case the number of possible labels is 2 (0/1 or -1/1), the problem is binary classification, and Hivemall's train_classifier()
function enables you to build binary classifiers. Binary Classification demonstrates how to use the function.
Another type of classification problems is multi-class classification. This task assumes that the number of possible labels is more than 2. We need to use different functions for the multi-class problems, and our news20 and iris tutorials would be helpful.
Here, we briefly explain about how prediction model is constructed.
First and foremost, we represent input and output for prediction models as follows:
For a set of samples $$(\mathbf{x}_1, y_1), (\mathbf{x}_2, y_2), \cdots, (\mathbf{x}_n, y_n)$$, the goal of prediction algorithms is to find a weight vector (i.e., parameters) $$\mathbf{w}$$ by minimizing the following error:
$$ E(\mathbf{w}) := \frac{1}{n} \sum_{i=1}^{n} L(\mathbf{w}; \mathbf{x}_i, y_i) + \lambda R(\mathbf{w}) $$
In the above formulation, there are two auxiliary functions we have to know:
($$\lambda$$ is a small value which controls the effect of regularization function.)
Eventually, minimizing the function $$E(\mathbf{w})$$ can be implemented by the SGD technique as described before, and $$\mathbf{w}$$ itself is used as a “model” for future prediction.
Interestingly, depending on a choice of loss and regularization function, prediction model you obtained will behave differently; even if one combination could work as a classifier, another choice might be appropriate for regression.
Below we list possible options for train_regressor
and train_classifier
, and this is the reason why these two functions are the most flexible in Hivemall:
Loss function: -loss
, -loss_function
train_regressor
train_classifier
Regularization function: -reg
, -regularization
Additionally, there are several variants of the SGD technique, and it is also configurable as:
-opt
, -optimizer
Note
Option values are case insensitive and you can use
sgd
orrda
, orhuberloss
in lower-case letters.
Furthermore, optimizer offers to set auxiliary options such as:
-iter
, -iterations
[default: 10]-cv_rate
, -convergence_rate
[default: 0.005]-iter
option.-mini_batch
, -mini_batch_size
[default: 1]For details of available options, following queries might be helpful to list all of them:
select train_regressor(array(), 0, '-help'); select train_classifier(array(), 0, '-help');
In practice, you can try different combinations of the options in order to achieve higher prediction accuracy.