Package org.neo4j.gds.ml.splitting
Class DirectedEdgeSplitter
java.lang.Object
org.neo4j.gds.ml.splitting.EdgeSplitter
org.neo4j.gds.ml.splitting.DirectedEdgeSplitter
-
Nested Class Summary
Nested classes/interfaces inherited from class org.neo4j.gds.ml.splitting.EdgeSplitter
EdgeSplitter.SplitResult -
Field Summary
Fields inherited from class org.neo4j.gds.ml.splitting.EdgeSplitter
POSITIVE, RELATIONSHIP_PROPERTY -
Constructor Summary
ConstructorsConstructorDescriptionDirectedEdgeSplitter(Optional<Long> maybeSeed, org.neo4j.gds.api.IdMap rootNodes, org.neo4j.gds.api.IdMap sourceLabels, org.neo4j.gds.api.IdMap targetLabels, org.neo4j.gds.RelationshipType selectedRelationshipType, org.neo4j.gds.RelationshipType remainingRelationshipType, int concurrency) -
Method Summary
Modifier and TypeMethodDescriptionprotected 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) protected longvalidPositiveRelationshipCandidateCount(org.neo4j.gds.api.Graph graph, com.carrotsearch.hppc.predicates.LongLongPredicate isValidNodePair) Methods inherited from class org.neo4j.gds.ml.splitting.EdgeSplitter
splitPositiveExamples
-
Constructor Details
-
DirectedEdgeSplitter
-
-
Method Details
-
validPositiveRelationshipCandidateCount
protected long validPositiveRelationshipCandidateCount(org.neo4j.gds.api.Graph graph, com.carrotsearch.hppc.predicates.LongLongPredicate isValidNodePair) - Specified by:
validPositiveRelationshipCandidateCountin classEdgeSplitter
-
positiveSampling
protected 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) - Specified by:
positiveSamplingin classEdgeSplitter
-