Skip to content

Commit 1225496

Browse files
committed
update comment
1 parent fabb194 commit 1225496

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

mllib/src/main/scala/org/apache/spark/ml/fpm/FPGrowth.scala

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -208,8 +208,8 @@ class FPGrowthModel private[ml] (
208208

209209
/**
210210
* Get association rules fitted by AssociationRules using the minConfidence. Returns a dataframe
211-
* with three fields, "antecedent", "consequent" and "confidence", where "antecedent" and
212-
* "consequent" are Array[T] and "confidence" is Double.
211+
* with four fields, "antecedent", "consequent", "confidence" and "support", where "antecedent"
212+
* and "consequent" are Array[T], "confidence" and "support" are Double.
213213
*/
214214
@Since("2.2.0")
215215
@transient lazy val associationRules: DataFrame = {
@@ -319,8 +319,8 @@ private[fpm] object AssociationRules {
319319
* @param freqCol column name for frequent itemsets count
320320
* @param numTrainingRecords count of training Dataset
321321
* @param minConfidence minimum confidence for the result association rules
322-
* @return a DataFrame("antecedent", "consequent", "confidence") containing the association
323-
* rules.
322+
* @return a DataFrame("antecedent", "consequent", "confidence", "support") containing the
323+
* association rules.
324324
*/
325325
def getAssociationRulesFromFP[T: ClassTag](
326326
dataset: Dataset[_],

0 commit comments

Comments
 (0)