fix: Update runner.py to fix a small bug (#282)

* fix: Update runner.py to fix a small bug

* fix CI
This commit is contained in:
Xu Yang
2024-09-20 14:03:46 +08:00
committed by GitHub
parent 0d36157a80
commit ed2130d8aa
+3 -1
View File
@@ -94,7 +94,9 @@ class KGFactorRunner(KGCachedRunner[KGFactorExperiment]):
factor_name="original features", factor_description="here is the original features", factor_formulation=""
)
org_data_path = Path(FACTOR_IMPLEMENT_SETTINGS.data_folder) / KAGGLE_IMPLEMENT_SETTING.competition / "valid.pkl"
org_data_path = (
Path(FACTOR_IMPLEMENT_SETTINGS.data_folder) / KAGGLE_IMPLEMENT_SETTING.competition / "X_valid.pkl"
)
with open(org_data_path, "rb") as f:
org_data = pickle.load(f)
feature_shape = org_data.shape[-1]