Package org.neo4j.gds.ml.models.linearregression

Class LinearRegressionTrainer

java.lang.Object

org.neo4j.gds.ml.models.linearregression.LinearRegressionTrainer

All Implemented Interfaces:

RegressorTrainer

public final class LinearRegressionTrainer
extends Object
implements RegressorTrainer

Constructor Summary

Constructors

Constructor	Description
LinearRegressionTrainer(int concurrency, LinearRegressionTrainConfig config, org.neo4j.gds.termination.TerminationFlag terminationFlag, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker, org.neo4j.gds.core.utils.progress.tasks.LogLevel messageLogLevel)

Method Summary

Modifier and Type	Method	Description
LinearRegressor	train(Features features, org.neo4j.gds.collections.ha.HugeDoubleArray targets, org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray trainSet)

Methods inherited from class java.lang.Object

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

Constructor Details

LinearRegressionTrainer

public LinearRegressionTrainer(int concurrency,
 LinearRegressionTrainConfig config,
 org.neo4j.gds.termination.TerminationFlag terminationFlag,
 org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker,
 org.neo4j.gds.core.utils.progress.tasks.LogLevel messageLogLevel)

Method Details

train

public LinearRegressor train(Features features,
 org.neo4j.gds.collections.ha.HugeDoubleArray targets,
 org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray trainSet)

Specified by:

train in interface RegressorTrainer