在自定义类中使用train_test_split时使用Typerror（单例数组…)

Question

TypeError：单个数组数组(<__Main__.Azhu_EmailClassifier_2对象位于0x000001D6E7A680D0>,DType=object)不能视为有效集合。

当我试图在自定义的AZHU_EmailClassifier_2类中运行train_test_split函数时，我遇到了这个错误。

我的班级：

class AZHU_EmailClassifier_2:
    import os
    import numpy as np
    import pandas as pd
    from sklearn.model_selection import train_test_split

    def __init__(self):
        pass
    
    def retrain_model(self, csv_file):
        
        MIN_ROW_NUMBER = 500
        TEST_SIZE = 0.25
        RANDOM_STATE = 42
        
        self.os.chdir(r"c:\LORI\PROJECTS\ALLIANZ\INCOMING_CHANNELS") # <---- a retraining file mappaja
        
        df=self.pd.read_excel(csv_file,error_bad_lines=False, header=None)
        
        df.dropna(axis=0,how='any', inplace=True)
             
        rows_no=df.shape[0]
        if rows_no<MIN_ROW_NUMBER:
            print("Insufficient number of rows (<35.000)! RETRAINING ABORTED")
            return None

        X=df[0]
        y=df[1]
        
        X_train, X_test, y_train, y_test=self.train_test_split(X,y)
        #X_train, X_test, y_train, y_test=self.train_test_split(X,y,test_size=TEST_SIZE, random_state=RANDOM_STATE, stratify=y)
        
        return X_train
                

~\Anaconda3\lib\site-packages\sklearn\utils\validation.py in indexable(*iterables)291“”“292 result=[_make_indexable(X)for X in iterables]-->293 check_consistent_length(*result)294返回结果295

~\Anaconda3\lib\site-packages\sklearn\utils\validation.py in check_consistent_length(*arrays)251“”“252-->253 lengths=[_num_samples(X)for X in arrays（如果X不是None）]254 uniques=np.unique（长度）255如果len(uniques)>1:

~\Anaconda3\lib\site-packages\sklearn\utils\validation.py in(.0)251“”252-->253长度=[_num_samples(X)for X in arrays（如果X不是None）]254 uniques=np.unique（长度）255如果len（uniques）>1:

~\Anaconda3\lib\site-packages\sklearn\utils\validation.py in_num_samples(x)194如果hasattr（x,'shape')和x.shape不是none：195如果len(x.shape)==0:-->196引发TypeError（“单例数组%r不能视为”197“有效集合。”%x)198#检查shape是否返回整数或默认值

TypeError：单个数组数组( ,DType=object)不能视为有效集合。

我不知道为什么它会抛出这个错误。你能给我指出正确的方向吗？感谢任何帮助！

Answer 1

出现此错误是因为在类内部导入了train_test_split，因此，train_test_split成为绑定方法而不是函数，并且每当调用该方法时，实例将作为第一个参数传递。这里有一个最小的例子可以重建这种情况

class test():
    
    from sklearn.model_selection import train_test_split
    
    def retrain_model(self):
        print(self.train_test_split)
        print(self.train_test_split())
    
test_instance = test()
test_instance.retrain_model()

运行此脚本后，将得到typeerror

TypeError: Singleton array array(<__main__.test object at 0x7ffa473ae438>, dtype=object) cannot be considered a valid collection.

而self.train_test_split在内存中的位置也是0x7FFA473AE438。

import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split

class AZHU_EmailClassifier_2():

    def __init__(self):
        pass
    
    def retrain_model(self,):
        
        MIN_ROW_NUMBER = 20
        TEST_SIZE = 0.25
        RANDOM_STATE = 42
                
        df = pd.DataFrame({0:np.linspace(1,100,100),1:np.random.rand(100)})
        X=df[0];y=df[1]
        X_train, X_test, y_train, y_test = train_test_split(X,y,test_size=TEST_SIZE,random_state=RANDOM_STATE)
        
        return X_train

test = AZHU_EmailClassifier_2()
test.retrain_model()