add decision tree ml model

Java-Cesco/Detecting_fraud_clicks/#10

add decision tree ml model
Java-Cesco/Detecting_fraud_clicks/#10
신은섭(Shin Eun Seop)
Commit 4d4027061ffee6dd06d2049cf6fb9b6c735cc880 4d402706 1 parent bb1e9781
Showing 4 changed files with 60 additions and 35 deletions
.gitignore
src/main/java/Aggregation.java → src/main/java/detact/Aggregation.java
src/main/java/MLModel.java → src/main/java/detact/ML/DecisionTree.java
src/main/java/Utill.java → src/main/java/detact/Utill.java
--- a/.gitignore
View file @4d40270
+++ b/.gitignore
View file @4d40270
@@ -75,3 +75,8 @@ fabric.properties
 # virtual machine crash logs, see http://www.java.com/en/download/help/error_hotspot.xml
 hs_err_pid*
+
+
+# datafile
+train.zip
+train.csv
\ No newline at end of file
--- a/src/main/java/Aggregation.java → src/main/java/detact/Aggregation.java
View file @4d40270
+++ b/src/main/java/Aggregation.java → src/main/java/detact/Aggregation.java
View file @4d40270
+package detact;
+
 import org.apache.spark.sql.Dataset;
 import org.apache.spark.sql.Row;
 import org.apache.spark.sql.SparkSession;
@@ -5,12 +7,13 @@ import org.apache.spark.sql.expressions.Window;
 import org.apache.spark.sql.expressions.WindowSpec;
 import static org.apache.spark.sql.functions.*;
-import static org.apache.spark.sql.functions.lit;
-import static org.apache.spark.sql.functions.when;
 public class Aggregation {
-    public static void main(String[] args) throws Exception {
+    public static String AGGREGATED_PATH = "agg_data";
+    public static String ORIGINAL_DATA_PATH = "train_sample.csv";
+
+    public static void main(String[] args) {
         //Create Session
         SparkSession spark = SparkSession
@@ -19,10 +22,10 @@ public class Aggregation {
                 .master("local")
                 .getOrCreate();
-        // Aggregation
+        // detact.Aggregation
         Aggregation agg = new Aggregation();
-        Dataset<Row> dataset = Utill.loadCSVDataSet("./train_sample.csv", spark);
+        Dataset<Row> dataset = Utill.loadCSVDataSet(Aggregation.ORIGINAL_DATA_PATH, spark);
         dataset = agg.changeTimestempToLong(dataset);
         dataset = agg.averageValidClickCount(dataset);
         dataset = agg.clickTimeDelta(dataset);
@@ -32,7 +35,7 @@ public class Aggregation {
         dataset.where("ip == '5348' and app == '19'").show(10);
         // Save to scv
-        Utill.saveCSVDataSet(dataset, "./agg_data");
+        Utill.saveCSVDataSet(dataset, Aggregation.AGGREGATED_PATH);
     }
     private Dataset<Row> changeTimestempToLong(Dataset<Row> dataset){
@@ -75,7 +78,7 @@ public class Aggregation {
                 .rangeBetween(Window.currentRow(),Window.currentRow()+600);
         Dataset<Row> newDF = dataset.withColumn("count_click_in_ten_mins",
-                (count("utc_click_time").over(w)).minus(1));    //TODO 본인것 포함할 것인지 정해야함.
+                (count("utc_click_time").over(w)).minus(1));  
         return newDF;
     }
--- a/src/main/java/MLModel.java → src/main/java/detact/ML/DecisionTree.java
View file @4d40270
+++ b/src/main/java/MLModel.java → src/main/java/detact/ML/DecisionTree.java
View file @4d40270
-import org.apache.spark.SparkConf;
+package detact.ML;
-import org.apache.spark.api.java.JavaRDD;
+
-import org.apache.spark.api.java.JavaSparkContext;
+import detact.Aggregation;
-import org.apache.spark.api.java.function.Function;
+import detact.Utill;
 import org.apache.spark.ml.Pipeline;
 import org.apache.spark.ml.PipelineModel;
 import org.apache.spark.ml.PipelineStage;
@@ -12,35 +12,47 @@ import org.apache.spark.ml.feature.VectorIndexerModel;
 import org.apache.spark.ml.regression.DecisionTreeRegressionModel;
 import org.apache.spark.ml.regression.DecisionTreeRegressor;
 import org.apache.spark.sql.Dataset;
-import org.apache.spark.sql.Encoders;
 import org.apache.spark.sql.Row;
-import org.apache.spark.sql.SQLContext;
+import org.apache.spark.sql.SparkSession;
-import scala.Serializable;
-
-import java.util.*;
-// ml
+// DecisionTree Model
-public class MapExample {
+public class DecisionTree {
     public static void main(String[] args) throws Exception {
-        // Automatically identify categorical features, and index them.
+        //Create Session
-        // Set maxCategories so features with > 4 distinct values are treated as continuous.
+        SparkSession spark = SparkSession
-        
+                .builder()
-        Aggregation agg = new Aggregation();
+                .appName("Detecting Fraud Clicks")
-        
+                .master("local")
-        agg.
+                .getOrCreate();
-        
+        
-        Dataset<Row> resultds = sqlContext.createDataFrame(result);
+        // load aggregated dataset
-
+        Dataset<Row> resultds = Utill.loadCSVDataSet(Aggregation.AGGREGATED_PATH, spark);
-        System.out.println("schema start");
+
-        resultds.printSchema();
+        // show Dataset schema
-        System.out.println("schema end");
+//        System.out.println("schema start");
+//        resultds.printSchema();
+//        String[] cols = resultds.columns();
+//        for (String col : cols) {
+//            System.out.println(col);
+//        }
+//        System.out.println("schema end");
         VectorAssembler assembler = new VectorAssembler()
-                .setInputCols(new String[]{"ip", "app", "device", "os", "channel", "clickInTenMins"})
+                .setInputCols(new String[]{
+                        "ip", 
+                        "app", 
+                        "device", 
+                        "os", 
+                        "channel", 
+                        "utc_click_time", 
+                        "avg_valid_click_count", 
+                        "click_time_delta",
+                        "count_click_in_ten_mins"
+                })
                 .setOutputCol("features");
         Dataset<Row> output = assembler.transform(resultds);
@@ -56,9 +68,11 @@ public class MapExample {
         Dataset<Row> trainingData = splits[0];
         Dataset<Row> testData = splits[1];
-        // Train a DecisionTree model.
+        // Train a detact.DecisionTreeionTree model.
         DecisionTreeRegressor dt = new DecisionTreeRegressor()
-                .setFeaturesCol("indexedFeatures").setLabelCol("attributed");
+                .setFeaturesCol("indexedFeatures")
+                .setLabelCol("is_attributed")
+                .setMaxDepth(10);
         // Chain indexer and tree in a Pipeline.
         Pipeline pipeline = new Pipeline()
@@ -71,11 +85,11 @@ public class MapExample {
         Dataset<Row> predictions = model.transform(testData);
         // Select example rows to display.
-        predictions.select("attributed", "features").show(5);
+        predictions.select("is_attributed", "features").show(5);
         // Select (prediction, true label) and compute test error.
         RegressionEvaluator evaluator = new RegressionEvaluator()
-                .setLabelCol("attributed")
+                .setLabelCol("is_attributed")
                 .setPredictionCol("prediction")
                 .setMetricName("rmse");
         double rmse = evaluator.evaluate(predictions);
@@ -86,4 +100,5 @@ public class MapExample {
         System.out.println("Learned regression tree model:\n" + treeModel.toDebugString());
     }
+    
 }
--- a/src/main/java/Utill.java → src/main/java/detact/Utill.java
View file @4d40270
+++ b/src/main/java/Utill.java → src/main/java/detact/Utill.java
View file @4d40270
+package detact;
+
 import org.apache.spark.sql.Dataset;
 import org.apache.spark.sql.Row;
 import org.apache.spark.sql.SparkSession;