cc.mallet.optimize.OptimizableCollection Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jcore-mallet-2.0.9 Show documentation

MALLET is a Java-based package for statistical natural language processing, document classification, clustering, topic modeling, information extraction, and other machine learning applications to text.

The newest version!

package cc.mallet.optimize;

import java.util.ArrayList;

import cc.mallet.types.MatrixOps;

public class OptimizableCollection {
	
	public class ByGradientValue implements Optimizable.ByGradientValue 
	{
		ArrayList optimizables;
		
		public ByGradientValue (Optimizable.ByGradientValue... ops) {
			optimizables = new ArrayList(ops.length);
			for (Optimizable.ByGradientValue o : ops)
				optimizables.add(o);
		}
		
		public void getValueGradient (double[] buffer) {
			double[] b2 = new double[buffer.length];
			for (Optimizable.ByGradientValue o : optimizables) {
				MatrixOps.setAll(b2, 0);
				o.getValueGradient(b2);
				MatrixOps.plusEquals(buffer, b2);
			}
		}
		
		public double getValue () {
			double ret = 0;
			for (Optimizable.ByGradientValue o : optimizables)
				ret += o.getValue();
			return ret;
		}

		// Here we rely on all optimizables pointing to the same set of parameters! 
		
		public int getNumParameters() {
			return optimizables.get(0).getNumParameters();
		}

		public double getParameter(int index) {
			return optimizables.get(0).getParameter(index);
		}

		public void getParameters(double[] buffer) {
			optimizables.get(0).getParameters(buffer);
		}

		public void setParameter(int index, double value) {
			optimizables.get(0).setParameter(index, value);
		}

		public void setParameters(double[] params) {
			optimizables.get(0).setParameters(params);
		}

		
	}

}