-
Notifications
You must be signed in to change notification settings - Fork 28.4k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[SPARK-10386] [MLlib] PrefixSpanModel supports save/load #10664
Conversation
Test build #49015 has finished for PR 10664 at commit
|
I'd like to get this into 2.0. Could you please fix the merge conflicts? Thanks! |
Test build #55193 has finished for PR 10664 at commit
|
@jkbradley I have resolved conflicts, this is ready for review now. Thanks! |
Taking a look now |
try { | ||
model.save(sc, path) | ||
val newModel = PrefixSpanModel.load(sc, path) | ||
val actual = newModel.freqSequences.collect() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I'd compare with the original model, not an expected result, here.
This looks good; I just had 1 comment |
Test build #55719 has finished for PR 10664 at commit
|
LGTM |
* If the directory already exists, this method throws an exception. | ||
*/ | ||
@Since("2.0.0") | ||
override def save(sc: SparkContext, path: String): Unit = { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Is sc used ?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Not here, but it's needed for other algorithms (with local representations).
PrefixSpanModel
supportssave/load
. It's similar with #9267.cc @jkbradley