From a7998b7bc667dd40b1ed75904107fcb1f40f1ee7 Mon Sep 17 00:00:00 2001
From: O2Dyokii <33623823+O2Dyokii@users.noreply.github.com>
Date: Fri, 21 Dec 2018 23:43:49 +0900
Subject: [PATCH] Update DCN.py

---
 RS-tf/DCN/DCN.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/RS-tf/DCN/DCN.py b/RS-tf/DCN/DCN.py
index 4743abc..794bc1a 100644
--- a/RS-tf/DCN/DCN.py
+++ b/RS-tf/DCN/DCN.py
@@ -129,6 +129,7 @@ def preprocessing(data):
 
 
 if __name__ == "__main__":
+    #data from kaggle forest cover type dataset
     data = pd.read_csv("data/covtype.csv")
     X, y, inp_layer, inp_embed = feature_generate(data)
     
@@ -137,4 +138,4 @@ if __name__ == "__main__":
     test_index = list(set(range(X.shape[0])) - set(train_index))
 
     model = fit(inp_layer, inp_embed, X.iloc[train_index], y[train_index, :])
-    evaluate(X.iloc[test_index], y[test_index, :], model)
\ No newline at end of file
+    evaluate(X.iloc[test_index], y[test_index, :], model)
-- 
GitLab