diff --git a/RS-tf/DCN/DCN.py b/RS-tf/DCN/DCN.py index 4743abc9e7130f796f9f218b97f484233ac8e931..794bc1a5c4296574ae6614e1c98b6cb910a016d9 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)