Skip to content

Commit 8698c37

Browse files
Drop nonSerializable models on shutdown and exclude them from backup
Also changes the semantics of "failIfUnsupportedType" to include nonSerializable models when the reason is (for example) non serializable trainerMethod rather than model type. Co-Authored-By: Adam Schill Collberg <adam.schill.collberg@protonmail.com>
1 parent 46f1a3b commit 8698c37

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ml/ml-algo/src/main/java/org/neo4j/gds/ml/models/logisticregression/LogisticRegressionData.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,10 @@
3131
import org.neo4j.gds.ml.models.Classifier;
3232
import org.neo4j.gds.ml.models.TrainingMethod;
3333

34+
import java.io.Serializable;
35+
3436
@ValueClass
35-
public interface LogisticRegressionData extends Classifier.ClassifierData {
37+
public interface LogisticRegressionData extends Classifier.ClassifierData, Serializable {
3638

3739
Weights<Matrix> weights();
3840
Weights<Vector> bias();

0 commit comments

Comments
 (0)