PythonIntermediate

Choose K with cross-validation

Use the inputs age, weight_kg, resting_heart_rate and weekly_sessions, the answer programme, and train_test_split with test_size=0.25, random_state=42 and stratify on the answer.

For every odd K from 1 to 25, score make_pipeline(StandardScaler(), KNeighborsClassifier(n_neighbors=K)) with cross_val_score (cv=5) on the training part only. Store the K with the highest average score in result (if two tie, the smaller K).

Already run for you

import io
import pandas as pd

members = pd.read_csv(io.StringIO("""member_id,age,weight_kg,resting_heart_rate,weekly_sessions,programme
601,32,84,64,5,Strength
602,20,63,63,5,Yoga
603,37,78,82,3,Cardio
604,45,83,74,4,Cardio
605,38,78,64,6,Strength
606,37,100,84,4,Cardio
607,48,70,70,4,Yoga
608,37,78,83,4,Yoga
609,38,65,73,2,Yoga
610,49,56,73,3,Yoga
611,58,62,75,2,Yoga
612,45,75,81,3,Yoga
613,36,70,66,6,Strength
614,36,60,82,3,Yoga
615,26,68,68,4,Strength
616,26,67,63,5,Yoga
617,27,87,72,5,Strength
618,50,57,72,2,Yoga
619,34,94,82,5,Cardio
620,40,85,81,4,Cardio
621,38,83,71,6,Strength
622,26,63,71,6,Strength
623,64,61,81,4,Yoga
624,24,65,72,6,Strength
625,37,69,80,2,Yoga
626,25,82,83,5,Cardio
627,56,66,72,2,Yoga
628,21,66,71,6,Strength
629,22,75,67,4,Strength
630,24,62,65,4,Strength
631,25,65,70,4,Strength
632,29,70,71,4,Strength
633,36,105,85,5,Cardio
634,37,81,84,5,Cardio
635,31,73,66,4,Strength
636,29,97,81,4,Cardio
637,22,71,70,6,Strength
638,37,81,82,3,Cardio
639,31,79,84,4,Cardio
640,26,65,62,4,Strength
641,23,86,69,4,Strength
642,43,78,81,4,Yoga
643,42,86,78,5,Cardio
644,51,59,75,4,Yoga
645,36,80,84,5,Cardio
646,24,74,72,4,Strength
647,22,71,72,4,Strength
648,53,73,81,3,Yoga
649,32,75,74,6,Strength
650,32,78,73,4,Strength
651,36,55,80,3,Yoga
652,30,85,63,4,Strength
653,41,89,81,4,Cardio
654,31,82,76,5,Cardio
655,45,79,76,3,Yoga
656,54,65,79,2,Strength
657,22,72,72,4,Strength
658,29,77,70,6,Strength
659,29,66,72,5,Strength
660,59,65,79,4,Yoga
661,23,83,73,4,Strength
662,20,86,71,5,Strength
663,37,59,76,4,Yoga
664,27,79,80,5,Cardio
665,37,67,73,4,Strength
666,35,77,72,6,Strength
667,37,71,65,5,Strength
668,37,100,77,5,Cardio
669,36,76,68,5,Strength
670,49,75,81,3,Yoga
671,28,83,81,3,Cardio
672,37,79,67,6,Strength
673,34,73,72,4,Cardio
674,34,101,86,3,Cardio
675,27,76,72,6,Strength
676,26,78,66,5,Strength
677,40,77,68,4,Strength
678,39,80,66,5,Strength
679,32,77,71,6,Strength
680,27,82,72,5,Strength
681,29,87,74,5,Cardio
682,29,101,80,4,Cardio
683,28,92,72,3,Strength
684,35,74,74,5,Strength
685,52,66,77,3,Yoga
686,37,56,75,3,Yoga
687,30,96,86,5,Cardio
688,25,101,77,4,Cardio
689,31,63,68,6,Strength
690,32,87,65,5,Strength
691,44,70,80,4,Yoga
692,35,78,71,6,Strength
693,34,80,78,4,Cardio
694,27,69,67,5,Strength
695,35,70,70,4,Strength
696,64,79,72,2,Yoga
697,32,80,68,4,Strength
698,27,94,73,4,Cardio
699,25,100,74,4,Cardio
700,21,78,70,5,Strength
701,43,85,73,4,Cardio
702,39,86,66,5,Strength
703,38,68,67,5,Strength
704,31,90,79,4,Cardio
705,38,72,77,2,Yoga
706,38,69,65,4,Strength
707,54,69,74,3,Yoga
708,63,72,78,2,Yoga
709,61,63,78,2,Yoga
710,31,81,67,4,Strength
711,57,63,79,4,Yoga
712,32,102,80,5,Yoga
713,56,64,72,2,Yoga
714,37,81,83,5,Cardio
715,38,82,76,5,Cardio
716,39,73,69,6,Strength
717,20,64,62,5,Strength
718,44,74,82,2,Yoga
719,41,97,86,4,Yoga
720,47,73,77,3,Yoga
721,34,87,63,5,Strength
722,31,93,80,5,Cardio
723,30,69,68,6,Strength
724,46,86,82,4,Cardio
725,34,106,82,3,Cardio
726,35,79,78,4,Cardio
727,44,77,72,3,Yoga
728,38,72,79,3,Yoga
729,27,104,81,5,Cardio
730,45,79,77,3,Yoga
731,39,85,85,3,Cardio
732,26,86,70,4,Strength
733,65,65,81,4,Yoga
734,29,83,77,5,Cardio
735,38,66,73,2,Yoga
736,29,80,69,6,Strength
737,28,100,81,3,Cardio
738,28,86,78,5,Cardio
739,59,70,83,4,Cardio
740,45,69,74,2,Yoga
741,29,69,67,6,Strength
742,25,80,68,5,Strength
743,40,95,76,4,Cardio
744,41,94,81,3,Cardio
745,32,81,64,4,Strength
746,63,58,74,2,Yoga
747,29,81,78,4,Cardio
748,46,78,71,3,Yoga
749,64,56,82,4,Strength
750,58,71,75,3,Yoga
751,26,76,63,5,Strength
752,51,66,80,2,Yoga
753,64,63,76,4,Yoga
754,38,73,79,3,Yoga
755,39,76,66,6,Strength
756,53,70,75,2,Yoga
757,44,64,71,4,Yoga
758,26,84,74,4,Cardio
759,33,84,62,4,Strength
760,26,85,85,5,Cardio
761,56,73,80,2,Yoga
762,28,104,73,5,Cardio
763,43,99,76,5,Cardio
764,46,105,80,5,Cardio
765,42,105,75,3,Cardio
766,30,63,70,5,Strength
767,37,66,72,2,Yoga
768,63,73,72,3,Yoga
769,46,75,72,2,Yoga
770,39,63,65,4,Strength
771,27,83,71,6,Strength
772,63,56,81,3,Yoga
773,32,66,63,4,Strength
774,24,74,71,6,Strength
775,27,73,64,5,Strength
776,28,85,84,3,Cardio
777,44,69,72,4,Cardio
778,25,71,69,6,Yoga
779,58,58,81,4,Yoga
780,29,92,84,4,Cardio"""))

Store your final answer in a variable named result.

Hints(0/2)

Each hint gives a little more help. Try on your own first.

Solution

Unlocks after you solve it or after 3 submissions

Notebook · scikit-learnRun and Submit check all cells from the top

Shift + Enter30 seconds limit per cell

[ ]

Run your code to see the output here.

Already run for you

import io import pandas as pd members = pd.read_csv(io.StringIO("""member_id,age,weight_kg,resting_heart_rate,weekly_sessions,programme 601,32,84,64,5,Strength 602,20,63,63,5,Yoga 603,37,78,82,3,Cardio 604,45,83,74,4,Cardio 605,38,78,64,6,Strength 606,37,100,84,4,Cardio 607,48,70,70,4,Yoga 608,37,78,83,4,Yoga 609,38,65,73,2,Yoga 610,49,56,73,3,Yoga 611,58,62,75,2,Yoga 612,45,75,81,3,Yoga 613,36,70,66,6,Strength 614,36,60,82,3,Yoga 615,26,68,68,4,Strength 616,26,67,63,5,Yoga 617,27,87,72,5,Strength 618,50,57,72,2,Yoga 619,34,94,82,5,Cardio 620,40,85,81,4,Cardio 621,38,83,71,6,Strength 622,26,63,71,6,Strength 623,64,61,81,4,Yoga 624,24,65,72,6,Strength 625,37,69,80,2,Yoga 626,25,82,83,5,Cardio 627,56,66,72,2,Yoga 628,21,66,71,6,Strength 629,22,75,67,4,Strength 630,24,62,65,4,Strength 631,25,65,70,4,Strength 632,29,70,71,4,Strength 633,36,105,85,5,Cardio 634,37,81,84,5,Cardio 635,31,73,66,4,Strength 636,29,97,81,4,Cardio 637,22,71,70,6,Strength 638,37,81,82,3,Cardio 639,31,79,84,4,Cardio 640,26,65,62,4,Strength 641,23,86,69,4,Strength 642,43,78,81,4,Yoga 643,42,86,78,5,Cardio 644,51,59,75,4,Yoga 645,36,80,84,5,Cardio 646,24,74,72,4,Strength 647,22,71,72,4,Strength 648,53,73,81,3,Yoga 649,32,75,74,6,Strength 650,32,78,73,4,Strength 651,36,55,80,3,Yoga 652,30,85,63,4,Strength 653,41,89,81,4,Cardio 654,31,82,76,5,Cardio 655,45,79,76,3,Yoga 656,54,65,79,2,Strength 657,22,72,72,4,Strength 658,29,77,70,6,Strength 659,29,66,72,5,Strength 660,59,65,79,4,Yoga 661,23,83,73,4,Strength 662,20,86,71,5,Strength 663,37,59,76,4,Yoga 664,27,79,80,5,Cardio 665,37,67,73,4,Strength 666,35,77,72,6,Strength 667,37,71,65,5,Strength 668,37,100,77,5,Cardio 669,36,76,68,5,Strength 670,49,75,81,3,Yoga 671,28,83,81,3,Cardio 672,37,79,67,6,Strength 673,34,73,72,4,Cardio 674,34,101,86,3,Cardio 675,27,76,72,6,Strength 676,26,78,66,5,Strength 677,40,77,68,4,Strength 678,39,80,66,5,Strength 679,32,77,71,6,Strength 680,27,82,72,5,Strength 681,29,87,74,5,Cardio 682,29,101,80,4,Cardio 683,28,92,72,3,Strength 684,35,74,74,5,Strength 685,52,66,77,3,Yoga 686,37,56,75,3,Yoga 687,30,96,86,5,Cardio 688,25,101,77,4,Cardio 689,31,63,68,6,Strength 690,32,87,65,5,Strength 691,44,70,80,4,Yoga 692,35,78,71,6,Strength 693,34,80,78,4,Cardio 694,27,69,67,5,Strength 695,35,70,70,4,Strength 696,64,79,72,2,Yoga 697,32,80,68,4,Strength 698,27,94,73,4,Cardio 699,25,100,74,4,Cardio 700,21,78,70,5,Strength 701,43,85,73,4,Cardio 702,39,86,66,5,Strength 703,38,68,67,5,Strength 704,31,90,79,4,Cardio 705,38,72,77,2,Yoga 706,38,69,65,4,Strength 707,54,69,74,3,Yoga 708,63,72,78,2,Yoga 709,61,63,78,2,Yoga 710,31,81,67,4,Strength 711,57,63,79,4,Yoga 712,32,102,80,5,Yoga 713,56,64,72,2,Yoga 714,37,81,83,5,Cardio 715,38,82,76,5,Cardio 716,39,73,69,6,Strength 717,20,64,62,5,Strength 718,44,74,82,2,Yoga 719,41,97,86,4,Yoga 720,47,73,77,3,Yoga 721,34,87,63,5,Strength 722,31,93,80,5,Cardio 723,30,69,68,6,Strength 724,46,86,82,4,Cardio 725,34,106,82,3,Cardio 726,35,79,78,4,Cardio 727,44,77,72,3,Yoga 728,38,72,79,3,Yoga 729,27,104,81,5,Cardio 730,45,79,77,3,Yoga 731,39,85,85,3,Cardio 732,26,86,70,4,Strength 733,65,65,81,4,Yoga 734,29,83,77,5,Cardio 735,38,66,73,2,Yoga 736,29,80,69,6,Strength 737,28,100,81,3,Cardio 738,28,86,78,5,Cardio 739,59,70,83,4,Cardio 740,45,69,74,2,Yoga 741,29,69,67,6,Strength 742,25,80,68,5,Strength 743,40,95,76,4,Cardio 744,41,94,81,3,Cardio 745,32,81,64,4,Strength 746,63,58,74,2,Yoga 747,29,81,78,4,Cardio 748,46,78,71,3,Yoga 749,64,56,82,4,Strength 750,58,71,75,3,Yoga 751,26,76,63,5,Strength 752,51,66,80,2,Yoga 753,64,63,76,4,Yoga 754,38,73,79,3,Yoga 755,39,76,66,6,Strength 756,53,70,75,2,Yoga 757,44,64,71,4,Yoga 758,26,84,74,4,Cardio 759,33,84,62,4,Strength 760,26,85,85,5,Cardio 761,56,73,80,2,Yoga 762,28,104,73,5,Cardio 763,43,99,76,5,Cardio 764,46,105,80,5,Cardio 765,42,105,75,3,Cardio 766,30,63,70,5,Strength 767,37,66,72,2,Yoga 768,63,73,72,3,Yoga 769,46,75,72,2,Yoga 770,39,63,65,4,Strength 771,27,83,71,6,Strength 772,63,56,81,3,Yoga 773,32,66,63,4,Strength 774,24,74,71,6,Strength 775,27,73,64,5,Strength 776,28,85,84,3,Cardio 777,44,69,72,4,Cardio 778,25,71,69,6,Yoga 779,58,58,81,4,Yoga 780,29,92,84,4,Cardio"""))