Java PolynomialCurveFitter.create方法代码示例

import org.apache.commons.math3.fitting.PolynomialCurveFitter; //导入方法依赖的package包/类
private double getSlope(Hashtable<Integer,Double> points)
{
    WeightedObservedPoints obs=new WeightedObservedPoints();
    Enumeration e=points.keys();
    int key;
    while (e.hasMoreElements())
    {
        key=(int)e.nextElement();
        obs.add(key,points.get(key));
    }
    PolynomialCurveFitter fitter=PolynomialCurveFitter.create(1);
    double[] coeff=fitter.fit(obs.toList());

    BigDecimal bd = new BigDecimal(coeff[1]);
    BigDecimal slope=bd.setScale(3,BigDecimal.ROUND_HALF_UP);
    return slope.doubleValue()*10;
}
 

import org.apache.commons.math3.fitting.PolynomialCurveFitter; //导入方法依赖的package包/类
public void fit(String path) throws IOException {
	final WeightedObservedPoints obs = new WeightedObservedPoints();
	try(BufferedReader br = new BufferedReader(new InputStreamReader(new FileInputStream(new File(path))))) {
		String line = br.readLine();
		while (line != null) {
			String[] point = line.split(" ");
			double x = Double.parseDouble(point[0]);
			double y = Double.parseDouble(point[1]);
			obs.add(x, y);
			line = br.readLine();
		}
	} catch (IOException e) {
		throw e;
	}
	final PolynomialCurveFitter fitter = PolynomialCurveFitter.create(5);
	final double[] coeff = fitter.fit(obs.toList());
	StringBuilder res = new StringBuilder();
	for (int i = coeff.length - 1; i >= 0; i--) {
		res.append(coeff[i]).append(":");
	}
	System.out.println(res.substring(0, res.length() - 1));
}
 

import org.apache.commons.math3.fitting.PolynomialCurveFitter; //导入方法依赖的package包/类
/**
 * TODO
 * 
 * @return
 */
public double process() {
	Collection<WeightedObservedPoint> points = new LinkedList<WeightedObservedPoint>();

	// TODO We might need to change the behavior due to the fact that cell that are not in a track might still be in the metaXml file
	for (MIFrame frame : forest.getMetaxml().getAllFrames()) {
		// For the growth rate to be correct we need elapsed time in hours
		double elapsedTime = frame.getElapsedTime() / 60;
		double cellNumber = (double) frame.getAllCellsInFrame().size();

		// The points are uniformly weighted
		points.add(new WeightedObservedPoint(1d, elapsedTime, Math.log(cellNumber)));
	}

	final AbstractCurveFitter fitter = PolynomialCurveFitter.create(1);
	double[] coeffs = fitter.fit(points);

	growthRate = coeffs[1];

	return growthRate;
}
 

import org.apache.commons.math3.fitting.PolynomialCurveFitter; //导入方法依赖的package包/类
/**
 * TODO
 * 
 * @return
 */
public double processNew() {
	Collection<WeightedObservedPoint> points = new LinkedList<WeightedObservedPoint>();

	// TODO We might need to change the behavior due to the fact that cell that are not in a track might still be in the metaXml file
	for (MIFrame frame : forest.getMetaxml().getAllFrames()) {
		// For the growth rate to be correct we need elapsed time in hours
		double elapsedTime = frame.getElapsedTime() / 60;
		double cellNumber = (double) frame.getAllCellsInFrame().size();

		// The points are uniformly weighted
		points.add(new WeightedObservedPoint(elapsedTime/forest.getMetaxml().getExperimentDuration(), elapsedTime, Math.log(cellNumber)));
	}

	final AbstractCurveFitter fitter = PolynomialCurveFitter.create(1);
	double[] coeffs = fitter.fit(points);
	
	factor = coeffs[0];
	growthRate = coeffs[1];

	return growthRate;
}
 

import org.apache.commons.math3.fitting.PolynomialCurveFitter; //导入方法依赖的package包/类
public static PolynomialFunction getStageEstimationFunction(
		List<Double> trainingSizes, List<Long> trainingDurations, int degree)
		throws SmallDataException {

	if (trainingSizes.size() <= degree)
		throw new SmallDataException();

	PolynomialCurveFitter fitter = PolynomialCurveFitter.create(degree);
	List<WeightedObservedPoint> points = new ArrayList<>();

	for (int i = 0; i < trainingSizes.size(); i++)
		points.add(new WeightedObservedPoint(1, trainingSizes.get(i),
				new Long(trainingDurations.get(i)).doubleValue()));

	return new PolynomialFunction(fitter.fit(points));

}
 

import org.apache.commons.math3.fitting.PolynomialCurveFitter; //导入方法依赖的package包/类
public static void detrend(List<double[]> set, int detrendPolynomial) {
	PolynomialCurveFitter p = PolynomialCurveFitter.create(detrendPolynomial);
	WeightedObservedPoints wop = new WeightedObservedPoints();
	for(int i = 0; i < set.size(); i++) {
		wop.add(set.get(i)[0], set.get(i)[1]);
	}
	double[] coeff = p.fit(wop.toList());
	for(int h = 0; h < set.size(); h++) {
		double val = set.get(h)[0];
		double off = 0;
		for(int i = detrendPolynomial; i >= 0; i--) {
			off += coeff[i] * Math.pow(val, i);
		}
		set.set(h, new double[]{set.get(h)[0], set.get(h)[1]-off});
	}
}
 

import org.apache.commons.math3.fitting.PolynomialCurveFitter; //导入方法依赖的package包/类
private boolean isLinearCoeffDiff(double[] input, int centerIndex) {
	double xStep = 2.0 / (input.length + 1);

	WeightedObservedPoints firstHalf = new WeightedObservedPoints();
	WeightedObservedPoints secondHalf = new WeightedObservedPoints();

	for (int i = 0; i < input.length; i++) {
		if (i <= centerIndex) {
			firstHalf.add(i * xStep, input[i]);
		}

		if (i >= centerIndex) {
			secondHalf.add((i - centerIndex) * xStep, input[i]);
		}
	}

	PolynomialCurveFitter fitter = PolynomialCurveFitter.create(1);

	double[] firstHalfCoeffs = fitter.fit(firstHalf.toList());
	double[] secondHalfCoeffs = fitter.fit(secondHalf.toList());

	double firstHalfCoeff = firstHalfCoeffs[1];
	double secondHalfCoeff = secondHalfCoeffs[1];

	return firstHalfCoeff * secondHalfCoeff < 0;
}
 

import org.apache.commons.math3.fitting.PolynomialCurveFitter; //导入方法依赖的package包/类
private double[] fitCurve(
	final List<ValuePair<DoubleType, DoubleType>> pairs)
{
	if (!allValuesFinite(pairs)) {
		return new double[2];
	}
	final WeightedObservedPoints points = toWeightedObservedPoints(pairs);
	final PolynomialCurveFitter curveFitter = PolynomialCurveFitter.create(1);
	return curveFitter.fit(points.toList());
}
 

import org.apache.commons.math3.fitting.PolynomialCurveFitter; //导入方法依赖的package包/类
/**
 * Get fitting coefficients for N degree polynomial.
 * @param x
 * @param y
 * @param degree
 * @param reverseCoefficients
 * @return Fitting coefficients.
 */
public double[] fitPolynomialToData(double[] x, double[] y, int degree, boolean reverseCoefficients) {
    WeightedObservedPoints obs = new WeightedObservedPoints();
    for (int i = 0; i < x.length; i++) {
        obs.add(1, x[i], y[i]);
    }
    PolynomialCurveFitter fitter = PolynomialCurveFitter.create(degree);
    double[] coefficients = fitter.fit(obs.toList());
    if (reverseCoefficients) {
        return reverseDoublePrimitiveArray(coefficients);
    } else {
        return coefficients;
    }
}
 

import org.apache.commons.math3.fitting.PolynomialCurveFitter; //导入方法依赖的package包/类
@Override
protected AbstractCurveFitter createFitter() {
    return PolynomialCurveFitter.create(degree);
}
 

import org.apache.commons.math3.fitting.PolynomialCurveFitter; //导入方法依赖的package包/类
public static long estimateStageDuration(List<Double> trainingSizes,
		List<Long> trainingDurations, double newSize, int degree) {

	PolynomialCurveFitter fitter = PolynomialCurveFitter.create(degree);
	List<WeightedObservedPoint> points = new ArrayList<>();

	for (int i = 0; i < trainingSizes.size(); i++)
		points.add(new WeightedObservedPoint(1, trainingSizes.get(i),
				new Long(trainingDurations.get(i)).doubleValue()));

	PolynomialFunction function = new PolynomialFunction(fitter.fit(points));

	return new Double(function.value(newSize)).longValue();

}