Skip to content

Commit 3d2d362

Browse files
committed
Fix fraud detection client imports
1 parent 95ffcb0 commit 3d2d362

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

research/fsi-fraud-detection/train/client.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,15 +25,14 @@
2525
from misc.data import all_model_parameters, flag, numerical_features, prepare_dataset
2626
from misc.data_io import load_csv_data_from_path, print_directory_tree, validate_data_features
2727
from misc.experiments import data_paths
28-
from model import SimpleNetwork
2928

3029
# Opacus for differential privacy
3130
from opacus import PrivacyEngine
3231
from opacus.validators import ModuleValidator
3332
from torch.optim.lr_scheduler import CosineAnnealingLR
3433
from torch.utils.data import DataLoader, TensorDataset, WeightedRandomSampler
35-
from train.utils import FocalLoss, str2bool
36-
from utils import MLflowCallback, compute_shapley_values, evaluate_on_test_datasets
34+
from train.model import SimpleNetwork
35+
from train.utils import FocalLoss, MLflowCallback, compute_shapley_values, evaluate_on_test_datasets, str2bool
3736

3837
# (1) import nvflare client API
3938
import nvflare.client as flare

0 commit comments

Comments
 (0)