Commit f6346bb3 authored by Joaquin Torres's avatar Joaquin Torres

Access to attribute names updated

parent d769c473
gen_train_data/input/ gen_train_data/output/post
gen_train_data/output/ gen_train_data/output/pre
EDA/input/17_abril.sav EDA/input/17_abril.sav
EDA/output/datasets EDA/output/datasets
\ No newline at end of file
...@@ -36,7 +36,7 @@ if __name__ == "__main__": ...@@ -36,7 +36,7 @@ if __name__ == "__main__":
# Setup # Setup
# -------------------------------------------------------------------------------------------------------- # --------------------------------------------------------------------------------------------------------
# Retrieve attribute names in order # Retrieve attribute names in order
attribute_names = list(np.load('../gen_train_data/output/attributes.npy', allow_pickle=True)) attribute_names = list(np.load('../EDA/output/feature_names/feature_names.npy', allow_pickle=True))
# Reading data # Reading data
data_dic = read_test_data(attribute_names) data_dic = read_test_data(attribute_names)
method_names = { method_names = {
......
...@@ -36,7 +36,7 @@ if __name__ == "__main__": ...@@ -36,7 +36,7 @@ if __name__ == "__main__":
# Setup # Setup
# -------------------------------------------------------------------------------------------------------- # --------------------------------------------------------------------------------------------------------
# Retrieve attribute names in order # Retrieve attribute names in order
attribute_names = list(np.load('../gen_train_data/output/attributes.npy', allow_pickle=True)) attribute_names = list(np.load('../EDA/output/feature_names/feature_names.npy', allow_pickle=True))
# Reading data # Reading data
data_dic = read_test_data(attribute_names) data_dic = read_test_data(attribute_names)
method_names = { method_names = {
......
...@@ -111,7 +111,7 @@ if __name__ == "__main__": ...@@ -111,7 +111,7 @@ if __name__ == "__main__":
# Setup # Setup
# -------------------------------------------------------------------------------------------------------- # --------------------------------------------------------------------------------------------------------
# Retrieve attribute names in order # Retrieve attribute names in order
attribute_names = list(np.load('../gen_train_data/output/attributes.npy', allow_pickle=True)) attribute_names = list(np.load('../EDA/output/feature_names/feature_names.npy', allow_pickle=True))
# Reading data # Reading data
data_dic = read_training_data(attribute_names) data_dic = read_training_data(attribute_names)
method_names = { method_names = {
......
...@@ -35,7 +35,7 @@ ...@@ -35,7 +35,7 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"# Retrieve attribute names in order\n", "# Retrieve attribute names in order\n",
"attribute_names = list(np.load('../gen_train_data/output/attributes.npy', allow_pickle=True))\n", "attribute_names = attribute_names = list(np.load('../EDA/output/feature_names/feature_names.npy', allow_pickle=True))\n",
"\n", "\n",
"# Load test data\n", "# Load test data\n",
"X_test_pre = np.load('../gen_train_data/output/pre/X_test_pre.npy', allow_pickle=True)\n", "X_test_pre = np.load('../gen_train_data/output/pre/X_test_pre.npy', allow_pickle=True)\n",
...@@ -71,8 +71,8 @@ ...@@ -71,8 +71,8 @@
" \"UNDER\": \"XGB\"\n", " \"UNDER\": \"XGB\"\n",
"}\n", "}\n",
"\n", "\n",
"soc_var_names = np.load('../EDA/output/soc_vars_names.npy', allow_pickle=True)\n", "soc_var_names = np.load('../EDA/output/feature_names/soc_vars_names.npy', allow_pickle=True)\n",
"ind_var_names = np.load('../EDA/output/ind_vars_names.npy', allow_pickle=True)" "ind_var_names = np.load('../EDA/output/feature_names/ind_vars_names.npy', allow_pickle=True)"
] ]
}, },
{ {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment