Github user mgaido91 commented on a diff in the pull request:
https://github.com/apache/spark/pull/22236#discussion_r212997377
--- Diff: mllib/src/main/scala/org/apache/spark/ml/fpm/FPGrowth.scala ---
@@ -326,6 +327,20 @@ object FPGrowthModel extends MLReadable[FPGrowthModel]
{
DefaultParamsWriter.saveMetadata(instance, path, sc)
val dataPath = new Path(path, "data").toString
instance.freqItemsets.write.parquet(dataPath)
+ val itemDataType =
instance.freqItemsets.schema(instance.getItemsCol).dataType match {
+ case ArrayType(et, _) => et
+ case other => throw new RuntimeException(s"Expected
${ArrayType.simpleString}, but got " +
--- End diff --
sure, I'll do, thanks.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
