summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorLoic Guegan <manzerbredes@mailbox.org>2022-11-09 16:53:34 +0100
committerLoic Guegan <manzerbredes@mailbox.org>2022-11-09 16:53:34 +0100
commit22391956eaf4053c24204af289168fd41356ce04 (patch)
treedc8bc6b2a5eb44fac593360c75935d41248c0dd3
parent4b31680b3adc0db25ed4eca98e04c4226440ca84 (diff)
Minor changes
-rw-r--r--analytics2/analysis.jl16
1 files changed, 16 insertions, 0 deletions
diff --git a/analytics2/analysis.jl b/analytics2/analysis.jl
new file mode 100644
index 0000000..1347d6e
--- /dev/null
+++ b/analytics2/analysis.jl
@@ -0,0 +1,16 @@
+using CSV, DataFrames, MLJ, Random
+
+data_csv=DataFrame(CSV.File("../CCGRID2022.csv"))
+data=groupby(data_csv,[:simkey,:wireless,:wakeupfor,:seed])
+data=combine(data,:nDataRcv => sum => :coverage, :energy => sum => :energy)
+
+# Building train/test sets
+train_idx=randsubseq(1:size(data,1),0.80)
+test_idx=setdiff(1:size(data,1),train_idx)
+train_set=data[train_idx,:]
+test_set=data[test_idx,:]
+
+# KNN
+knn=@load KNNClassifier
+y,X=unpack(data,==(:simkey))
+knn_machine=machine(knn(),select(X,Not(:wireless),Not(:seed)),select(data,:seed)) \ No newline at end of file