[R] train set 만들어주기

ISLR 5.4 Exercises 에서

5번 문제를 풀다가 배우게 된 내용이다.

 

validation 을 진행하기 위해 전체 데이터를

1) train 데이터와 2) validation 데이터로 나누고자 한다.

(test 데이터도 만들 수 있지만 여기서는 이렇게 두 가지만 사용하는 것이 가정이다.)

 

아래와 같이 만들어주면 된다.

train <- sample(dim(Default)[1], dim(Default)[1]/2)

• dim(Default): 데이터프레임 Default의 행, 열의 수를 반환한다.
  - 결과값은 10,000 / 4 가 나온다.
  
• dim(Default)[1]: 행, 열의 수 중 행을 일컫는다.
  - 여기서는 10,000
  - (참고) 파이썬과 같이 인덱싱이 0부터 시작하지 않고 1부터 시작한다.
• dim(Default)[1]/2: 10,000 나누기 2 이기 때문에 값이 5,000 이 된다.
• sample(A, B): 앞의 A 범위에서 B개를 무작위로 선택하라는 것이다.
  - 여기서는 10,000개의 범위에서 5,000개를 무작위로 선택하라는 것이고, 이것을 train 세트로 지정하였다.