Repository: spark Updated Branches: refs/heads/master 3ff81ad2d -> f7efda397
[SPARK-9959] [MLLIB] Association Rules Java Compatibility mengxr Author: Feynman Liang <fli...@databricks.com> Closes #8206 from feynmanliang/SPARK-9959-arules-java. Project: http://git-wip-us.apache.org/repos/asf/spark/repo Commit: http://git-wip-us.apache.org/repos/asf/spark/commit/f7efda39 Tree: http://git-wip-us.apache.org/repos/asf/spark/tree/f7efda39 Diff: http://git-wip-us.apache.org/repos/asf/spark/diff/f7efda39 Branch: refs/heads/master Commit: f7efda3975d46a8ce4fd720b3730127ea482560b Parents: 3ff81ad Author: Feynman Liang <fli...@databricks.com> Authored: Mon Aug 17 09:58:34 2015 -0700 Committer: Xiangrui Meng <m...@databricks.com> Committed: Mon Aug 17 09:58:34 2015 -0700 ---------------------------------------------------------------------- .../spark/mllib/fpm/AssociationRules.scala | 30 ++++++++++++++++++-- 1 file changed, 28 insertions(+), 2 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/spark/blob/f7efda39/mllib/src/main/scala/org/apache/spark/mllib/fpm/AssociationRules.scala ---------------------------------------------------------------------- diff --git a/mllib/src/main/scala/org/apache/spark/mllib/fpm/AssociationRules.scala b/mllib/src/main/scala/org/apache/spark/mllib/fpm/AssociationRules.scala index 72d0ea0..7f4de77 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/fpm/AssociationRules.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/fpm/AssociationRules.scala @@ -16,6 +16,7 @@ */ package org.apache.spark.mllib.fpm +import scala.collection.JavaConverters._ import scala.reflect.ClassTag import org.apache.spark.Logging @@ -95,8 +96,10 @@ object AssociationRules { * :: Experimental :: * * An association rule between sets of items. - * @param antecedent hypotheses of the rule - * @param consequent conclusion of the rule + * @param antecedent hypotheses of the rule. Java users should call [[Rule#javaAntecedent]] + * instead. + * @param consequent conclusion of the rule. Java users should call [[Rule#javaConsequent]] + * instead. * @tparam Item item type * * @since 1.5.0 @@ -108,6 +111,11 @@ object AssociationRules { freqUnion: Double, freqAntecedent: Double) extends Serializable { + /** + * Returns the confidence of the rule. + * + * @since 1.5.0 + */ def confidence: Double = freqUnion.toDouble / freqAntecedent require(antecedent.toSet.intersect(consequent.toSet).isEmpty, { @@ -115,5 +123,23 @@ object AssociationRules { s"A valid association rule must have disjoint antecedent and " + s"consequent but ${sharedItems} is present in both." }) + + /** + * Returns antecedent in a Java List. + * + * @since 1.5.0 + */ + def javaAntecedent: java.util.List[Item] = { + antecedent.toList.asJava + } + + /** + * Returns consequent in a Java List. + * + * @since 1.5.0 + */ + def javaConsequent: java.util.List[Item] = { + consequent.toList.asJava + } } } --------------------------------------------------------------------- To unsubscribe, e-mail: commits-unsubscr...@spark.apache.org For additional commands, e-mail: commits-h...@spark.apache.org