아래와 같이 predict 에서 에러가 납니다.

TypeError                                 Traceback (most recent call last)

/tmp/ipython-input-2218855363.py in <cell line: 0>()
      1 #9. 학습 시 사용하지 않은 데이터로 가격 예측
----> 2 y_pred = reg.predict(X_test.values)

7 frames

/usr/local/lib/python3.12/dist-packages/keras/src/saving/serialization_lib.py in _retrieve_class_or_fn(name, registered_name, module, obj_type, full_config, custom_objects)
    808                 )
    809 
--> 810     raise TypeError(
    811         f"Could not locate {obj_type} '{name}'. "
    812         "Make sure custom classes are decorated with "

TypeError: Could not locate class 'AddOneDimension'. Make sure custom classes are decorated with `@keras.saving.register_keras_serializable()`. Full object config: {'module': 'autokeras.preprocessors.common', 'class_name': 'AddOneDimension', 'config': {}, 'registered_name': 'AddOneDimension'}