Showing
3 changed files
with
30 additions
and
23 deletions
| ... | @@ -22,7 +22,7 @@ public class Aggregation { | ... | @@ -22,7 +22,7 @@ public class Aggregation { |
| 22 | // Aggregation | 22 | // Aggregation |
| 23 | Aggregation agg = new Aggregation(); | 23 | Aggregation agg = new Aggregation(); |
| 24 | 24 | ||
| 25 | - Dataset<Row> dataset = agg.loadCSVDataSet("./train_sample.csv", spark); | 25 | + Dataset<Row> dataset = Utill.loadCSVDataSet("./train_sample.csv", spark); |
| 26 | dataset = agg.changeTimestempToLong(dataset); | 26 | dataset = agg.changeTimestempToLong(dataset); |
| 27 | dataset = agg.averageValidClickCount(dataset); | 27 | dataset = agg.averageValidClickCount(dataset); |
| 28 | dataset = agg.clickTimeDelta(dataset); | 28 | dataset = agg.clickTimeDelta(dataset); |
| ... | @@ -31,16 +31,7 @@ public class Aggregation { | ... | @@ -31,16 +31,7 @@ public class Aggregation { |
| 31 | //test | 31 | //test |
| 32 | dataset.where("ip == '5348' and app == '19'").show(10); | 32 | dataset.where("ip == '5348' and app == '19'").show(10); |
| 33 | 33 | ||
| 34 | - agg.saveCSVDataSet(dataset, "./agg_data"); | 34 | + Utill.saveCSVDataSet(dataset, "./agg_data"); |
| 35 | - } | ||
| 36 | - | ||
| 37 | - | ||
| 38 | - private Dataset<Row> loadCSVDataSet(String path, SparkSession spark){ | ||
| 39 | - // Read SCV to DataSet | ||
| 40 | - return spark.read().format("csv") | ||
| 41 | - .option("inferSchema", "true") | ||
| 42 | - .option("header", "true") | ||
| 43 | - .load(path); | ||
| 44 | } | 35 | } |
| 45 | 36 | ||
| 46 | private Dataset<Row> changeTimestempToLong(Dataset<Row> dataset){ | 37 | private Dataset<Row> changeTimestempToLong(Dataset<Row> dataset){ |
| ... | @@ -87,12 +78,4 @@ public class Aggregation { | ... | @@ -87,12 +78,4 @@ public class Aggregation { |
| 87 | return newDF; | 78 | return newDF; |
| 88 | } | 79 | } |
| 89 | 80 | ||
| 90 | - private void saveCSVDataSet(Dataset<Row> dataset, String path){ | ||
| 91 | - // Read SCV to DataSet | ||
| 92 | - dataset.repartition(1) | ||
| 93 | - .write().format("csv") | ||
| 94 | - .option("inferSchema", "true") | ||
| 95 | - .option("header", "true") | ||
| 96 | - .save(path); | ||
| 97 | - } | ||
| 98 | } | 81 | } | ... | ... |
| ... | @@ -24,14 +24,15 @@ import java.util.*; | ... | @@ -24,14 +24,15 @@ import java.util.*; |
| 24 | 24 | ||
| 25 | public class MapExample { | 25 | public class MapExample { |
| 26 | 26 | ||
| 27 | - static SparkConf conf = new SparkConf().setMaster("local[*]").setAppName("Cesco"); | ||
| 28 | - static JavaSparkContext sc = new JavaSparkContext(conf); | ||
| 29 | - static SQLContext sqlContext = new SQLContext(sc); | ||
| 30 | - | ||
| 31 | public static void main(String[] args) throws Exception { | 27 | public static void main(String[] args) throws Exception { |
| 32 | 28 | ||
| 33 | // Automatically identify categorical features, and index them. | 29 | // Automatically identify categorical features, and index them. |
| 34 | // Set maxCategories so features with > 4 distinct values are treated as continuous. | 30 | // Set maxCategories so features with > 4 distinct values are treated as continuous. |
| 31 | + | ||
| 32 | + Aggregation agg = new Aggregation(); | ||
| 33 | + | ||
| 34 | + agg. | ||
| 35 | + | ||
| 35 | Dataset<Row> resultds = sqlContext.createDataFrame(result); | 36 | Dataset<Row> resultds = sqlContext.createDataFrame(result); |
| 36 | 37 | ||
| 37 | System.out.println("schema start"); | 38 | System.out.println("schema start"); | ... | ... |
src/main/java/Utill.java
0 → 100644
| 1 | +import org.apache.spark.sql.Dataset; | ||
| 2 | +import org.apache.spark.sql.Row; | ||
| 3 | +import org.apache.spark.sql.SparkSession; | ||
| 4 | + | ||
| 5 | +public class Utill { | ||
| 6 | + | ||
| 7 | + public static Dataset<Row> loadCSVDataSet(String path, SparkSession spark){ | ||
| 8 | + // Read SCV to DataSet | ||
| 9 | + return spark.read().format("csv") | ||
| 10 | + .option("inferSchema", "true") | ||
| 11 | + .option("header", "true") | ||
| 12 | + .load(path); | ||
| 13 | + } | ||
| 14 | + | ||
| 15 | + public static void saveCSVDataSet(Dataset<Row> dataset, String path){ | ||
| 16 | + // Read SCV to DataSet | ||
| 17 | + dataset.repartition(1) | ||
| 18 | + .write().format("csv") | ||
| 19 | + .option("inferSchema", "true") | ||
| 20 | + .option("header", "true") | ||
| 21 | + .save(path); | ||
| 22 | + } | ||
| 23 | +} |
-
Please register or login to post a comment