Github user mgaido91 commented on a diff in the pull request:
https://github.com/apache/spark/pull/22236#discussion_r213574108
--- Diff: mllib/src/main/scala/org/apache/spark/ml/fpm/FPGrowth.scala ---
@@ -217,7 +222,9 @@ object FPGrowth extends DefaultParamsReadable[FPGrowth]
{
@Experimental
class FPGrowthModel private[ml] (
@Since("2.2.0") override val uid: String,
- @Since("2.2.0") @transient val freqItemsets: DataFrame)
+ @Since("2.2.0") @transient val freqItemsets: DataFrame,
+ private val itemSupport: scala.collection.Map[Any, Double],
--- End diff --
Yes, `Map[_, Double]` seems to work too. I can change it if you prefer.
The `numTrainingRecords` is used because it needs to be stored (and it then
used when loading the model in order to recompute the `itemSupport`).
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
