SAMOA-34: Check for ensemble consistency

Project: http://git-wip-us.apache.org/repos/asf/incubator-samoa/repo
Commit: http://git-wip-us.apache.org/repos/asf/incubator-samoa/commit/9cdfb578
Tree: http://git-wip-us.apache.org/repos/asf/incubator-samoa/tree/9cdfb578
Diff: http://git-wip-us.apache.org/repos/asf/incubator-samoa/diff/9cdfb578

Branch: refs/heads/master
Commit: 9cdfb578876e71f32fef687e9ca5c12e66987be4
Parents: fc44099
Author: Gianmarco De Francisci Morales <[email protected]>
Authored: Thu Jun 18 16:29:50 2015 +0300
Committer: abifet <[email protected]>
Committed: Wed Jul 1 18:07:04 2015 +0800

----------------------------------------------------------------------
 .../classifiers/ensemble/BaggingDistributorProcessor.java        | 4 ++++
 1 file changed, 4 insertions(+)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/incubator-samoa/blob/9cdfb578/samoa-api/src/main/java/org/apache/samoa/learners/classifiers/ensemble/BaggingDistributorProcessor.java
----------------------------------------------------------------------
diff --git 
a/samoa-api/src/main/java/org/apache/samoa/learners/classifiers/ensemble/BaggingDistributorProcessor.java
 
b/samoa-api/src/main/java/org/apache/samoa/learners/classifiers/ensemble/BaggingDistributorProcessor.java
index 6c88d94..a49065b 100644
--- 
a/samoa-api/src/main/java/org/apache/samoa/learners/classifiers/ensemble/BaggingDistributorProcessor.java
+++ 
b/samoa-api/src/main/java/org/apache/samoa/learners/classifiers/ensemble/BaggingDistributorProcessor.java
@@ -34,6 +34,8 @@ import org.apache.samoa.learners.InstanceContentEvent;
 import org.apache.samoa.moa.core.MiscUtils;
 import org.apache.samoa.topology.Stream;
 
+import com.google.common.base.Preconditions;
+
 /**
  * The Class BaggingDistributorPE.
  */
@@ -58,6 +60,8 @@ public class BaggingDistributorProcessor implements Processor 
{
    * @return true, if successful
    */
   public boolean process(ContentEvent event) {
+    Preconditions.checkState(ensembleSize == ensembleStreams.length, 
String.format(
+        "Ensemble size ({}) and number of enseble streams ({}) do not match.", 
ensembleSize, ensembleStreams.length));
     InstanceContentEvent inEvent = (InstanceContentEvent) event;
 
     if (inEvent.getInstanceIndex() < 0) {

Reply via email to