Update DCN.py

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)
+    evaluate(X.iloc[test_index], y[test_index, :], model)
\ No newline at end of file