Package org.neo4j.gds.ml.pipeline.nodePipeline.regression

Class NodeRegressionToModelConverter

java.lang.Object

org.neo4j.gds.ml.pipeline.nodePipeline.regression.NodeRegressionToModelConverter

All Implemented Interfaces:

ResultToModelConverter<NodeRegressionTrainResult.NodeRegressionTrainPipelineResult,NodeRegressionTrainResult>

public class NodeRegressionToModelConverter
extends java.lang.Object
implements ResultToModelConverter<NodeRegressionTrainResult.NodeRegressionTrainPipelineResult,NodeRegressionTrainResult>

Constructor Summary

Constructors

Constructor	Description
NodeRegressionToModelConverter(NodeRegressionTrainingPipeline pipeline, NodeRegressionPipelineTrainConfig config)

Method Summary

Modifier and Type	Method	Description
NodeRegressionTrainResult.NodeRegressionTrainPipelineResult	toModel(NodeRegressionTrainResult trainResult, 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

NodeRegressionToModelConverter

public NodeRegressionToModelConverter(NodeRegressionTrainingPipeline pipeline,
                                      NodeRegressionPipelineTrainConfig config)

Method Detail

toModel

public NodeRegressionTrainResult.NodeRegressionTrainPipelineResult toModel(NodeRegressionTrainResult trainResult,
                                                                           org.neo4j.gds.api.schema.GraphSchema originalSchema)

Specified by:

toModel in interface ResultToModelConverter<NodeRegressionTrainResult.NodeRegressionTrainPipelineResult,NodeRegressionTrainResult>