전체가 61개의 열로 되어 있고, 마지막 열이 광물의 종류를 나타냅니다. 일반 암석일 경우 0, 광석일 경우 1로 표시되어 있습니다. 첫 번째 열부터 60번째 열까지는 음파 주파수의 에너지를 0에서 1 사이의 숫자로 표시하고 있습니다. 이제 일반 암석과 광석이 각각 몇 개나 포함되어 있는지 알아보겠습니다.

df[60].value_counts()

실행 결과

1    111
0     97
Name: 60, dtype: int64

광석이 111개, 일반 암석이 97개, 총 208개의 샘플이 준비되어 있는 것을 알 수 있습니다. 이제 다음과 같이 1~60번째 열을 X 변수에 저장하고 광물의 종류는 y로 저장하겠습니다.

X = df.iloc[:,0:60]
y = df.iloc[:,60]

이후 앞서 했던 그대로 딥러닝을 실행하겠습니다.