Class UserInputNegativeSampler

java.lang.Object
org.neo4j.gds.ml.negativeSampling.UserInputNegativeSampler
All Implemented Interfaces:
NegativeSampler

public class UserInputNegativeSampler extends Object implements NegativeSampler
  • Constructor Details

    • UserInputNegativeSampler

      public UserInputNegativeSampler(org.neo4j.gds.api.Graph negativeExampleGraph, double testTrainFraction, Optional<Long> randomSeed, Collection<org.neo4j.gds.NodeLabel> sourceLabels, Collection<org.neo4j.gds.NodeLabel> targetLabels)
  • Method Details

    • produceNegativeSamples

      public void produceNegativeSamples(org.neo4j.gds.core.loading.construction.RelationshipsBuilder testSetBuilder, org.neo4j.gds.core.loading.construction.RelationshipsBuilder trainSetBuilder)
      Specified by:
      produceNegativeSamples in interface NegativeSampler