Skip to content

Commit

Permalink
formatting fix
Browse files Browse the repository at this point in the history
Signed-off-by: kta-intel <[email protected]>
  • Loading branch information
kta-intel committed Nov 18, 2024
1 parent b346b24 commit 809b69b
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion openfl-workspace/xgb_higgs/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
modin[all]
scikit-learn
xgboost
modin[all]
4 changes: 2 additions & 2 deletions openfl-workspace/xgb_higgs/src/dataloader.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ def load_Higgs(data_path, **kwargs):
Load the Higgs dataset from CSV files.
The dataset is expected to be in two CSV files: 'train.csv' and 'valid.csv'.
The first column in each file represents the labels, and the remaining
The first column in each file represents the labels, and the remaining
columns represent the features.
Args:
Expand All @@ -57,4 +57,4 @@ def load_Higgs(data_path, **kwargs):
X_valid = valid_data.iloc[:, 1:].values
y_valid = valid_data.iloc[:, 0].values

return X_train, y_train, X_valid, y_valid
return X_train, y_train, X_valid, y_valid
4 changes: 2 additions & 2 deletions openfl/federated/data/loader_xgb.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from math import ceil
pass

import numpy as np
import xgboost as xgb
Expand Down Expand Up @@ -60,7 +60,7 @@ def get_valid_data_size(self):
int: The total number of validation samples.
"""
return self.X_valid.shape[0]

def get_dmatrix(self, X, y):
"""Returns the DMatrix for the given data.
Expand Down

0 comments on commit 809b69b

Please sign in to comment.