Package org.neo4j.gds.ml.pipeline.linkPipeline.train

Class LinkPredictionRelationshipSampler

java.lang.Object

org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionRelationshipSampler

public class LinkPredictionRelationshipSampler
extends java.lang.Object

Constructor Summary

Constructors

Constructor	Description
LinkPredictionRelationshipSampler(org.neo4j.gds.api.GraphStore graphStore, LinkPredictionSplitConfig splitConfig, LinkPredictionTrainConfig trainConfig, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker, org.neo4j.gds.core.utils.TerminationFlag terminationFlag)

Method Summary

Modifier and Type	Method	Description
static org.neo4j.gds.core.utils.mem.MemoryEstimation	estimateNegativeSampling(java.lang.String relationshipType, double testFraction, double trainFraction, double negativeSamplingRatio, java.util.Optional<java.lang.String> negativeRelationshipType)
static org.neo4j.gds.core.utils.mem.MemoryEstimation	estimatePositiveRelations(java.lang.String relationshipType, double testFraction, double trainFraction, java.util.Optional<java.lang.String> relationshipWeight)
void	splitAndSampleRelationships(java.util.Optional<java.lang.String> relationshipWeightProperty)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

LinkPredictionRelationshipSampler

public LinkPredictionRelationshipSampler(org.neo4j.gds.api.GraphStore graphStore,
                                         LinkPredictionSplitConfig splitConfig,
                                         LinkPredictionTrainConfig trainConfig,
                                         org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker,
                                         org.neo4j.gds.core.utils.TerminationFlag terminationFlag)

Method Detail

splitAndSampleRelationships

public void splitAndSampleRelationships(java.util.Optional<java.lang.String> relationshipWeightProperty)

estimatePositiveRelations

public static org.neo4j.gds.core.utils.mem.MemoryEstimation estimatePositiveRelations(java.lang.String relationshipType,
                                                                                      double testFraction,
                                                                                      double trainFraction,
                                                                                      java.util.Optional<java.lang.String> relationshipWeight)

estimateNegativeSampling

public static org.neo4j.gds.core.utils.mem.MemoryEstimation estimateNegativeSampling(java.lang.String relationshipType,
                                                                                     double testFraction,
                                                                                     double trainFraction,
                                                                                     double negativeSamplingRatio,
                                                                                     java.util.Optional<java.lang.String> negativeRelationshipType)