org.apache.mahout.classifier.OnlineLearner Maven / Gradle / Ivy

/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.mahout.classifier;

import org.apache.mahout.math.Vector;

import java.io.Closeable;

/**
 * The simplest interface for online learning algorithms.
 */
public interface OnlineLearner extends Closeable {
  /**
   * Updates the model using a particular target variable value and a feature vector.
   * 
   * There may an assumption that if multiple passes through the training data are necessary, then
   * the training examples will be presented in the same order.  This is because the order of
   * training examples may be used to assign records to different data splits for evaluation by
   * cross-validation.  Without the order invariance, records might be assigned to training and test
   * splits and error estimates could be seriously affected.
   * 

   * If re-ordering is necessary, then using the alternative API which allows a tracking key to be
   * added to the training example can be used.
   *
   * @param actual   The value of the target variable.  This value should be in the half-open
   *                 interval [0..n) where n is the number of target categories.
   * @param instance The feature vector for this example.
   */
  void train(int actual, Vector instance);

  /**
   * Updates the model using a particular target variable value and a feature vector.
   * 

   * There may an assumption that if multiple passes through the training data are necessary that
   * the tracking key for a record will be the same for each pass and that there will be a
   * relatively large number of distinct tracking keys and that the low-order bits of the tracking
   * keys will not correlate with any of the input variables.  This tracking key is used to assign
   * training examples to different test/training splits.
   * 

   * Examples of useful tracking keys include id-numbers for the training records derived from
   * a database id for the base table from the which the record is derived, or the offset of
   * the original data record in a data file.
   *
   * @param trackingKey The tracking key for this training example.
   * @param groupKey     An optional value that allows examples to be grouped in the computation of
   * the update to the model.
   * @param actual   The value of the target variable.  This value should be in the half-open
   *                 interval [0..n) where n is the number of target categories.
   * @param instance The feature vector for this example.
   */
  void train(long trackingKey, String groupKey, int actual, Vector instance);

  /**
   * Updates the model using a particular target variable value and a feature vector.
   * 

   * There may an assumption that if multiple passes through the training data are necessary that
   * the tracking key for a record will be the same for each pass and that there will be a
   * relatively large number of distinct tracking keys and that the low-order bits of the tracking
   * keys will not correlate with any of the input variables.  This tracking key is used to assign
   * training examples to different test/training splits.
   * 
   * Examples of useful tracking keys include id-numbers for the training records derived from
   * a database id for the base table from the which the record is derived, or the offset of
   * the original data record in a data file.
   *
   * @param trackingKey The tracking key for this training example.
   * @param actual   The value of the target variable.  This value should be in the half-open
   *                 interval [0..n) where n is the number of target categories.
   * @param instance The feature vector for this example.
   */
  void train(long trackingKey, int actual, Vector instance);

  /**
   * Prepares the classifier for classification and deallocates any temporary data structures.
   *
   * An online classifier should be able to accept more training after being closed, but
   * closing the classifier may make classification more efficient.
   */
  @Override
  void close();
}