com.actelion.research.orbit.imageAnalysis.utils.GmmSemiDemo Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of orbit-image-analysis Show documentation
Show all versions of orbit-image-analysis Show documentation
Orbit, a versatile image analysis software for biological image-based quantification
/*
* Orbit, a versatile image analysis software for biological image-based quantification.
* Copyright (C) 2009 - 2018 Idorsia Pharmaceuticals Ltd., Hegenheimermattweg 91, CH-4123 Allschwil, Switzerland.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see .
*
*/
package com.actelion.research.orbit.imageAnalysis.utils;
import org.apache.commons.math3.distribution.MixtureMultivariateNormalDistribution;
import org.apache.commons.math3.distribution.MultivariateNormalDistribution;
import java.util.Arrays;
import java.util.Random;
/**
* Creates two multivariate gaussian clusters with some known labels.
* Uses GmmSemi to apply semi-supervised learning (clustering with some known labels) to detect the clusters.
*
* Manuel, 2018
*
*/
public class GmmSemiDemo {
public static void main(String[] args) {
// generate data (two clusters)
double[] means1 = new double[]{8,13}; // cluster center 0
double[][] covar1 = new double[][] {
{ 1.5, 0.8 },
{ 0.8, 1.5 }
};
double[] means2 = new double[]{5,10}; // cluster center 1
double[][] covar2 = new double[][] {
{ 1, 0.6 },
{ 0.6, 1 }
};
MultivariateNormalDistribution mns1 = new MultivariateNormalDistribution(means1,covar1);
MultivariateNormalDistribution mns2 = new MultivariateNormalDistribution(means2,covar2);
Random random = new Random();
int n = 10000; // #data points
int g = 2; // #clusters
double[][] data = new double[n][g];
double[] trueLabels = new double[n];
double[] labels = new double[n];
double pLabel = 0.3d; // probability that label is known
double p0 = 0.5d; // probability for datapoint belonging to cluster0
for (int i=0; i