package org.apache.mahout.clustering.dirichlet;
/**
* 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.
*/
import java.awt.BasicStroke;
import java.awt.Graphics;
import java.awt.Graphics2D;
import org.apache.mahout.clustering.dirichlet.models.Model;
import org.apache.mahout.clustering.dirichlet.models.NormalModel;
import org.apache.mahout.clustering.dirichlet.models.NormalModelDistribution;
import org.apache.mahout.matrix.DenseVector;
import org.apache.mahout.matrix.Vector;
class DisplayNDirichlet extends DisplayDirichlet {
public DisplayNDirichlet() {
initialize();
this.setTitle("Dirichlet Process Clusters - Normal Distribution (>"
+ (int) (significance * 100) + "% of population)");
}
private static final long serialVersionUID = 1L;
public void paint(Graphics g) {
super.plotSampleData(g);
Graphics2D g2 = (Graphics2D) g;
Vector dv = new DenseVector(2);
int i = result.size() - 1;
for (Model<Vector>[] models : result) {
g2.setStroke(new BasicStroke(i == 0 ? 3 : 1));
g2.setColor(colors[Math.min(colors.length - 1, i--)]);
for (Model<Vector> m : models) {
NormalModel mm = (NormalModel) m;
dv.assign(mm.sd * 3);
if (isSignificant(mm))
plotEllipse(g2, mm.mean, dv);
}
}
}
public static void main(String[] args) {
UncommonDistributions.init("Mahout=Hadoop+ML".getBytes());
generateSamples();
generateResults();
new DisplayNDirichlet();
}
static void generateResults() {
DisplayDirichlet.generateResults(new NormalModelDistribution());
}
}