빅데이터 분석기사

Data must be 1-dimensional 오류

유방울 2023. 6. 14. 00:41

Data must be 1-dimensional 오류 의심할 부분

빅데이터 분석기사 시험응시 시 2유형 문제풀이를 가정했을 때이다.

코드는 빅데이터 분석기사 4회기출을 기반하고 있다.

1. minmax 를 먼저 하고 train, submission으로 나누기

ndarray 배열이므로 iloc 사용 안 하고 -> 슬라이싱 사용하기 

def get_data(dfX,Y):
    X = dfX.drop(columns='ID')
    scaler = MinMaxScaler()
    X_scaled = scaler.fit_transform(X)
    X1_use = X_scaled[:6665, :]
    X1_submission = X_scaled[6665:, :]
    Y1 = Y['Segmentation']
    print(X1_use.shape, Y1.shape, X1_submission.shape)
    return X1_use, X1_submission, Y1


2. predict 가 아니라 predict_prob를 한 경우

predict_prob 메서드는 2차원으로 반환됨

+ 2차원으로 반환되는 메서드를 사용하고 있을 경우 이 부분은 좀 더 공부해야 알 수 있을 거 같다.