Package org.neo4j.gds.ml.splitting
Class EdgeSplitter
java.lang.Object
org.neo4j.gds.ml.splitting.EdgeSplitter
- Direct Known Subclasses:
DirectedEdgeSplitter,UndirectedEdgeSplitter
-
Nested Class Summary
Nested Classes -
Field Summary
Fields -
Method Summary
Modifier and TypeMethodDescriptionprotected abstract voidpositiveSampling(org.neo4j.gds.api.Graph graph, org.neo4j.gds.core.loading.construction.RelationshipsBuilder selectedRelsBuilder, org.neo4j.gds.api.RelationshipWithPropertyConsumer remainingRelsConsumer, org.apache.commons.lang3.mutable.MutableLong selectedRelCount, org.apache.commons.lang3.mutable.MutableLong remainingRelCount, long nodeId, com.carrotsearch.hppc.predicates.LongLongPredicate isValidNodePair, org.apache.commons.lang3.mutable.MutableLong positiveSamplesRemaining, org.apache.commons.lang3.mutable.MutableLong candidateEdgesRemaining) splitPositiveExamples(org.neo4j.gds.api.Graph graph, double holdoutFraction, Optional<String> remainingRelPropertyKey) protected abstract longvalidPositiveRelationshipCandidateCount(org.neo4j.gds.api.Graph graph, com.carrotsearch.hppc.predicates.LongLongPredicate isValidNodePair)
-
Field Details
-
POSITIVE
public static final double POSITIVE- See Also:
-
RELATIONSHIP_PROPERTY
- See Also:
-
-
Method Details
-
splitPositiveExamples
public EdgeSplitter.SplitResult splitPositiveExamples(org.neo4j.gds.api.Graph graph, double holdoutFraction, Optional<String> remainingRelPropertyKey) -
positiveSampling
protected abstract void positiveSampling(org.neo4j.gds.api.Graph graph, org.neo4j.gds.core.loading.construction.RelationshipsBuilder selectedRelsBuilder, org.neo4j.gds.api.RelationshipWithPropertyConsumer remainingRelsConsumer, org.apache.commons.lang3.mutable.MutableLong selectedRelCount, org.apache.commons.lang3.mutable.MutableLong remainingRelCount, long nodeId, com.carrotsearch.hppc.predicates.LongLongPredicate isValidNodePair, org.apache.commons.lang3.mutable.MutableLong positiveSamplesRemaining, org.apache.commons.lang3.mutable.MutableLong candidateEdgesRemaining) -
validPositiveRelationshipCandidateCount
protected abstract long validPositiveRelationshipCandidateCount(org.neo4j.gds.api.Graph graph, com.carrotsearch.hppc.predicates.LongLongPredicate isValidNodePair)
-