Package org.neo4j.gds.ml.pipeline.nodePipeline.classification.train

Class NodeClassificationToModelConverter

java.lang.Object

org.neo4j.gds.ml.pipeline.nodePipeline.classification.train.NodeClassificationToModelConverter

All Implemented Interfaces:

ResultToModelConverter<NodeClassificationTrainResult.NodeClassificationModelResult,NodeClassificationTrainResult>

public class NodeClassificationToModelConverter
extends java.lang.Object
implements ResultToModelConverter<NodeClassificationTrainResult.NodeClassificationModelResult,NodeClassificationTrainResult>

Constructor Summary

Constructors

Constructor	Description
NodeClassificationToModelConverter(NodeClassificationTrainingPipeline pipeline, NodeClassificationPipelineTrainConfig config)

Method Summary

Modifier and Type	Method	Description
NodeClassificationTrainResult.NodeClassificationModelResult	toModel(NodeClassificationTrainResult result, org.neo4j.gds.api.schema.GraphSchema originalSchema)

Methods inherited from class java.lang.Object

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

Constructor Detail

NodeClassificationToModelConverter

public NodeClassificationToModelConverter(NodeClassificationTrainingPipeline pipeline,
                                          NodeClassificationPipelineTrainConfig config)

Method Detail

toModel

public NodeClassificationTrainResult.NodeClassificationModelResult toModel(NodeClassificationTrainResult result,
                                                                           org.neo4j.gds.api.schema.GraphSchema originalSchema)

Specified by:

toModel in interface ResultToModelConverter<NodeClassificationTrainResult.NodeClassificationModelResult,NodeClassificationTrainResult>