KMeansTask.java 1.7 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/***********************************************************************************************************************
 *
 * Copyright (C) 2010-2014 by the Stratosphere project (http://stratosphere.eu)
 *
 * Licensed 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.
 *
 **********************************************************************************************************************/

Y
Yingjun Wu 已提交
16
package eu.stratosphere.streaming.examples.iterative.kmeans;
17

Y
Yingjun Wu 已提交
18
import eu.stratosphere.api.java.tuple.Tuple1;
19 20 21
import eu.stratosphere.streaming.api.invokable.UserTaskInvokable;
import eu.stratosphere.streaming.api.streamrecord.StreamRecord;

Y
Yingjun Wu 已提交
22
public class KMeansTask extends UserTaskInvokable {
23

Y
Yingjun Wu 已提交
24 25 26 27 28
	private static final long serialVersionUID = 1L;
	private StreamRecord outRecord = new StreamRecord(new Tuple1<String>());
	private double[] point=null;
	public KMeansTask(int dimension){
		point = new double[dimension];
29
	}
Y
Yingjun Wu 已提交
30
	
31 32 33
	@Override
	public void invoke(StreamRecord record) throws Exception {
		// TODO Auto-generated method stub
Y
Yingjun Wu 已提交
34 35 36 37 38 39
		String[] pointStr = record.getString(0, 0).split(" ");
		for(int i=0; i<pointStr.length; ++i){
			point[i]=Double.valueOf(pointStr[i]);
		}
		outRecord.setString(0, record.getString(0, 0));
		emit(outRecord);
40
	}
Y
Yingjun Wu 已提交
41

42
}