টিএল; ডিআর: এর সাথে স্ট্র্যাটেইডশফ্লস্প্লিট ব্যবহার করুনtest_size=0.25
স্কাইকিট-লার্ন স্ট্র্যাফাইড বিভাজনের জন্য দুটি মডিউল সরবরাহ করে:
- স্ট্র্যাটেডিকেফোর্ড : এই মডিউলটি সরাসরি কে-ভাঁজ ক্রস-বৈধকরণ অপারেটর হিসাবে দরকারী: এটি
n_folds
প্রশিক্ষণ / পরীক্ষার সেটগুলি সেট করবে যে উভয় শ্রেণিতে সমান ভারসাম্যপূর্ণ।
এখানে কিছু কোড (সরাসরি উপরের ডকুমেন্টেশন থেকে)
>>> skf = cross_validation.StratifiedKFold(y, n_folds=2)
>>> len(skf)
2
>>> for train_index, test_index in skf:
... print("TRAIN:", train_index, "TEST:", test_index)
... X_train, X_test = X[train_index], X[test_index]
... y_train, y_test = y[train_index], y[test_index]
...
- স্ট্র্যাটেডশফ্লস্প্লিট : এই মডিউলটি একটি একক প্রশিক্ষণ / পরীক্ষার সেট তৈরি করে যা সমান ভারসাম্যযুক্ত (স্তরযুক্ত) শ্রেণিযুক্ত। মূলত আপনি এর সাথে এটিই চান
n_iter=1
। আপনি এখানে পরীক্ষার আকার উল্লেখ করতে পারেনtrain_test_split
কোড:
>>> sss = StratifiedShuffleSplit(y, n_iter=1, test_size=0.5, random_state=0)
>>> len(sss)
1
>>> for train_index, test_index in sss:
... print("TRAIN:", train_index, "TEST:", test_index)
... X_train, X_test = X[train_index], X[test_index]
... y_train, y_test = y[train_index], y[test_index]
>>>