Remove check
This commit is contained in:
parent
732060d160
commit
769c07373c
@ -42,11 +42,6 @@ def run(gt, rasters, coords, remove, attributes, adjacency='4', filtering='direc
|
|||||||
X_raw = np.moveaxis(np.array(list(eap.vectorize())), 0, -1)
|
X_raw = np.moveaxis(np.array(list(eap.vectorize())), 0, -1)
|
||||||
y_raw = gti
|
y_raw = gti
|
||||||
|
|
||||||
if np.isnan(X_raw).any():
|
|
||||||
raise Exception('Found {} Nan on {}'.format(np.isnan(X_raw).sum(),
|
|
||||||
coordsi))
|
|
||||||
|
|
||||||
|
|
||||||
# Remove unwanted label X, y
|
# Remove unwanted label X, y
|
||||||
lbl = np.ones_like(y_raw, dtype=np.bool)
|
lbl = np.ones_like(y_raw, dtype=np.bool)
|
||||||
for l in remove if remove else []:
|
for l in remove if remove else []:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user