Class DirectedEdgeSplitter

java.lang.Object
org.neo4j.gds.ml.splitting.EdgeSplitter
org.neo4j.gds.ml.splitting.DirectedEdgeSplitter

public class DirectedEdgeSplitter extends EdgeSplitter
  • 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

    Constructors
    Constructor
    Description
    DirectedEdgeSplitter(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 Type
    Method
    Description
    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)
     
    protected long
    validPositiveRelationshipCandidateCount(org.neo4j.gds.api.Graph graph, com.carrotsearch.hppc.predicates.LongLongPredicate isValidNodePair)
     

    Methods inherited from class org.neo4j.gds.ml.splitting.EdgeSplitter

    splitPositiveExamples

    Methods inherited from class java.lang.Object

    clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
  • Constructor Details

    • DirectedEdgeSplitter

      public DirectedEdgeSplitter(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 Details

    • validPositiveRelationshipCandidateCount

      protected long validPositiveRelationshipCandidateCount(org.neo4j.gds.api.Graph graph, com.carrotsearch.hppc.predicates.LongLongPredicate isValidNodePair)
      Specified by:
      validPositiveRelationshipCandidateCount in class EdgeSplitter
    • 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:
      positiveSampling in class EdgeSplitter