NVE / avalanche_ml

Applying machine learning in the Norwegian Avalanche warning Service
MIT License
2 stars 1 forks source link

Bug when removing regression targets #19

Open kmunve opened 4 years ago

kmunve commented 4 years ago

Building in the following code in main.py

# For example, do this to remove all "REAL" regression targets:
labeled_data.label.drop("REAL", axis=1, level=0, inplace=True)
labeled_data.pred.drop("REAL", axis=1, level=0, inplace=True)

Results in an error in aggregatedata.py

.../avalanche_ml_widforss/modeling/decion_trees.py
2020-09-23 12:48:01.063115: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
Reading csv
Training fold: 0
Traceback (most recent call last):
  File "C:/Users/kmu/PycharmProjects/avalanche_ml_widforss/modeling/decion_trees.py", line 60, in <module>
    bm.fit(training_data, epochs=80, verbose=1)
  File "C:\Users\kmu\PycharmProjects\avalanche_ml_widforss\machine.py", line 80, in fit
    self.dummies = labeled_data.to_dummies()['label']
  File "C:\Users\kmu\PycharmProjects\avalanche_ml_widforss\aggregatedata.py", line 839, in to_dummies
    for subprob in df["REAL"].columns.get_level_values(0).unique():
  File "C:\ProgramData\Anaconda3\envs\ml\lib\site-packages\pandas\core\frame.py", line 2799, in __getitem__
    return self._getitem_multilevel(key)
  File "C:\ProgramData\Anaconda3\envs\ml\lib\site-packages\pandas\core\frame.py", line 2849, in _getitem_multilevel
    loc = self.columns.get_loc(key)
  File "C:\ProgramData\Anaconda3\envs\ml\lib\site-packages\pandas\core\indexes\multi.py", line 2662, in get_loc
    loc = self._get_level_indexer(key, level=0)
  File "C:\ProgramData\Anaconda3\envs\ml\lib\site-packages\pandas\core\indexes\multi.py", line 2943, in _get_level_indexer
    raise KeyError(key)
KeyError: 'REAL'

Process finished with exit code 1