Skip to content

Commit da1d591

Browse files
authored
Scikit learn (#643)
* Replace sklearn with scikit-learn * Use underscore in python files * Lint fix
1 parent bf1050d commit da1d591

4 files changed

Lines changed: 9 additions & 4 deletions

File tree

miqa/learning/correlator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#!/usr/bin/env python3
22

33
import pandas as pd
4-
from sklearn.metrics import confusion_matrix
4+
from scikit_learn.metrics import confusion_matrix
55

66
df = pd.read_csv('M:/MIQA/data.csv') # manually converted TRUE/FALSE into 1/0
77
print(f'count NaN: {df.isnull().sum().sum()}')

miqa/learning/nn_inference.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,12 @@
44
import itk
55
import monai
66
import numpy as np
7-
from sklearn.metrics import classification_report, confusion_matrix, mean_squared_error, r2_score
7+
from scikit_learn.metrics import (
8+
classification_report,
9+
confusion_matrix,
10+
mean_squared_error,
11+
r2_score,
12+
)
813
import torch
914
from torch.utils.data import DataLoader
1015
import torchio

miqa/learning/nn_training.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
)
2222
import numpy as np
2323
import pandas as pd
24-
from sklearn.metrics import confusion_matrix
24+
from scikit_learn.metrics import confusion_matrix
2525
import torch
2626
from torch.utils.data import DataLoader
2727
from torch.utils.tensorboard import SummaryWriter

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@
7373
'learning': [
7474
'itk>=5.3rc4',
7575
'monai',
76-
'sklearn',
76+
'scikit-learn',
7777
'torch',
7878
'torchio',
7979
'wandb',

0 commit comments

Comments
 (0)