Skip to content

Commit

Permalink
Create PredictiveAnalytics.java
Browse files Browse the repository at this point in the history
  • Loading branch information
KOSASIH authored Jul 25, 2024
1 parent f6fa566 commit 81e134c
Showing 1 changed file with 20 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
import org.apache.spark.ml.regression.LinearRegression;
import org.apache.spark.ml.regression.LinearRegressionModel;
import org.apache.spark.sql.Dataset;
import org.apache.spark.sql.Row;

public class PredictiveAnalytics {
public static void main(String[] args) {
// Load data
Dataset<Row> data = spark.read().format("csv").option("header", "true").load("data.csv");

// Split data into training and testing sets
Dataset<Row>[] splits = data.randomSplit(new double[]{0.7, 0.3}, 42);
Dataset<Row> trainingData = splits[0];
Dataset<Row> testingData = splits[1];

// Train a linear regression model
LinearRegression lr = new LinearRegression();
LinearRegressionModel model = lr.fit(trainingData);
}
}

0 comments on commit 81e134c

Please sign in to comment.