correct module "cross_validation" to "model_selection"

declare-lab · Oct 21, 2019 · c72e33a · c72e33a
1 parent 4a7b658
commit c72e33a
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 8 deletions.
diff --git a/CMN/IEMOCAP/cmn.py b/CMN/IEMOCAP/cmn.py
@@ -2,7 +2,7 @@
 import numpy as np
 import sys
 import tensorflow.contrib.rnn as rnn_cell
-from sklearn.cross_validation import train_test_split
+from sklearn.model_selection import train_test_split
 
 
 

diff --git a/CMN/IEMOCAP/utils_cmn.py b/CMN/IEMOCAP/utils_cmn.py
@@ -1,15 +1,15 @@
 import numpy as np
 import pandas as pd
 import pickle
-from sklearn import cross_validation, metrics
+from sklearn import model_selection, metrics
 
 TEXT_EMBEDDINGS = "./IEMOCAP/data/text/IEMOCAP_text_embeddings.pickle"
 VIDEO_EMBEDDINGS = "./IEMOCAP/data/video/IEMOCAP_video_features.pickle"
 AUDIO_EMBEDDINGS = "./IEMOCAP/data/audio/IEMOCAP_audio_features.pickle"
 
 trainID = pickle.load(open("./IEMOCAP/data/trainID.pkl",'rb'), encoding="latin1")
 testID = pickle.load(open("./IEMOCAP/data/testID.pkl",'rb'), encoding="latin1")
-valID,_ = cross_validation.train_test_split(testID, test_size=.4, random_state=1227)
+valID,_ = model_selection.train_test_split(testID, test_size=.4, random_state=1227)
 # valID = testID
 
 transcripts, labels, own_historyID, other_historyID, own_historyID_rank, other_historyID_rank = pickle.load(open("./IEMOCAP/data/dataset.pkl",'rb'), encoding="latin1")

diff --git a/CMN/train_iemocap.py b/CMN/train_iemocap.py
@@ -5,7 +5,7 @@
 from IEMOCAP.utils_cmn import *
 from IEMOCAP.cmn import *
 import os
-from sklearn import cross_validation, metrics
+from sklearn import model_selection, metrics
 from sklearn.metrics import classification_report
 from sklearn.metrics import confusion_matrix
 

diff --git a/ICON/IEMOCAP/model.py b/ICON/IEMOCAP/model.py
@@ -2,7 +2,7 @@
 import numpy as np
 import sys
 import tensorflow.contrib.rnn as rnn_cell
-from sklearn.cross_validation import train_test_split
+from sklearn.model_selection import train_test_split
 
 
 

diff --git a/ICON/IEMOCAP/utils.py b/ICON/IEMOCAP/utils.py
@@ -1,15 +1,15 @@
 import numpy as np
 import pandas as pd
 import pickle
-from sklearn import cross_validation, metrics
+from sklearn import model_selection, metrics
 
 TEXT_EMBEDDINGS = "./IEMOCAP/data/text/IEMOCAP_text_embeddings.pickle"
 VIDEO_EMBEDDINGS = "./IEMOCAP/data/video/IEMOCAP_video_features.pickle"
 AUDIO_EMBEDDINGS = "./IEMOCAP/data/audio/IEMOCAP_audio_features.pickle"
 
 trainID = pickle.load(open("./IEMOCAP/data/trainID.pkl",'rb'), encoding="latin1")
 testID = pickle.load(open("./IEMOCAP/data/testID.pkl",'rb'), encoding="latin1")
-valID,_ = cross_validation.train_test_split(testID, test_size=.4, random_state=1227)
+valID,_ = model_selection.train_test_split(testID, test_size=.4, random_state=1227)
 # valID = testID
 
 transcripts, labels, own_historyID, other_historyID, own_historyID_rank, other_historyID_rank = pickle.load(open("./IEMOCAP/data/dataset.pkl",'rb'), encoding="latin1")

diff --git a/ICON/train_iemocap.py b/ICON/train_iemocap.py
@@ -4,7 +4,7 @@
 from IEMOCAP.utils import *
 from IEMOCAP.model import *
 import os
-from sklearn import cross_validation, metrics
+from sklearn import model_selection, metrics
 from sklearn.metrics import classification_report
 from sklearn.metrics import confusion_matrix