如何检查训练集或测试集是否为空? - Python 代码示例
可以使用以下方法检查训练集或测试集是否为空:
- 使用Python中的len()函数检查数据集的长度是否为0,如果为0则为空。
例如:
if len(train_set) == 0:
print('训练集为空')
if len(test_set) == 0:
print('测试集为空')
- 使用Pandas库读取数据集,然后使用DataFrame.empty属性检查数据集是否为空。
例如:
import pandas as pd
train_set = pd.read_csv('train.csv')
test_set = pd.read_csv('test.csv')
if train_set.empty:
print('训练集为空')
if test_set.empty:
print('测试集为空')
- 使用Numpy库读取数据集,然后使用ndarray.size属性检查数据集中的元素数量是否为0,如果为0则为空。
例如:
import numpy as np
train_set = np.loadtxt('train.txt')
test_set = np.loadtxt('test.txt')
if train_set.size == 0:
print('训练集为空')
if test_set.size == 0:
print('测试集为空')
原文地址: https://www.cveoy.top/t/topic/owNQ 著作权归作者所有。请勿转载和采集!