KMeans报错

import glob
import pandas as pd
from sklearn.cluster import KMeans
import os

上面是在代码中导入的包，执行发现报错。

UserWarning: KMeans is known to have a memory leak on Windows with MKL, when there are less chunks than available threads. You can avoid it by setting the environment variable OMP_NUM_THREADS=1.

根据报错信息设置环境变量后。

import glob
import pandas as pd
from sklearn.cluster import KMeans
import os
os.environ['OMP_NUM_THREADS'] = '1'

运行发现仍然报错，经排查，需要在导入KMeans之前，导入os并设置环境变量。

import os
os.environ['OMP_NUM_THREADS'] = '1'
import glob
import pandas as pd
from sklearn.cluster import KMeans

这样设置后，运行成功。