Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 17 additions & 0 deletions multivariate_view/app/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,9 +43,17 @@ def __init__(self, server=None):
dest="preprocess",
action='store_true',
)
self.server.cli.add_argument(
"--ignore-fields",
help="Fields to ignore from the dataset",
default=[],
nargs='+',
type=str,
)

args, _ = self.server.cli.parse_known_args()
self.enable_preprocessing = args.preprocess
self.ignore_fields = args.ignore_fields
file_to_load = args.data
if file_to_load is None:
EXAMPLE_DATA_DIR.mkdir(parents=True, exist_ok=True)
Expand Down Expand Up @@ -85,6 +93,15 @@ def __init__(self, server=None):
def load_data(self, file_to_load):
header, data = load_dataset(Path(file_to_load))

for key in self.ignore_fields:
if key in header:
index_to_remove = header.index(key)
data = np.delete(data, index_to_remove, axis=3)
else:
print(
f"Warning: field {key} is not part of the provided dataset.\nThe available fields are {header}"
)

self.state.component_labels = header

if self.enable_preprocessing:
Expand Down