# p r 1   
 
 
 
 i m p o r t   p a n d a s   a s   p d 
 
 i m p o r t   n u m p y   a s   n p 
 
 f r o m   s k l e a r n . m o d e l _ s e l e c t i o n   i m p o r t   t r a i n _ t e s t _ s p l i t 
 
 f r o m   s k l e a r n . p r e p r o c e s s i n g   i m p o r t   S t a n d a r d S c a l e r ,   L a b e l E n c o d e r 
 
 
 
 d a t a   =   p d . r e a d _ c s v ( " H e a r t . c s v " ) 
 
 p r i n t ( "   D a t a   l o a d e d   s u c c e s s f u l l y ! " ) 
 
 p r i n t ( d a t a ) 
 
 
 
 
 
 
 
 d a t a   =   d a t a . d r o p _ d u p l i c a t e s ( ) 
 
 p r i n t ( " \ n M i s s i n g   v a l u e s   b e f o r e   c l e a n i n g : \ n " ,   d a t a . i s n u l l ( ) . s u m ( ) ) 
 
 
 
 
 
 
 
 d a t a [ ' C a ' ]   =   d a t a [ ' C a ' ] . f i l l n a ( d a t a [ ' C a ' ] . m e a n ( ) ) 
 
 d a t a [ ' T h a l ' ]   =   d a t a [ ' T h a l ' ] . f i l l n a ( d a t a [ ' T h a l ' ] . m o d e ( ) [ 0 ] ) 
 
 d a t a . i s n u l l ( ) . s u m ( ) 
 
 
 
 
 
 
 
 f o r   c o l   i n   d a t a . c o l u m n s : 
 
         i f   d a t a [ c o l ] . d t y p e   = =   ' o b j e c t ' : 
 
                 l e   =   L a b e l E n c o d e r ( ) 
 
                 d a t a [ c o l ]   =   l e . f i t _ t r a n s f o r m ( d a t a [ c o l ] ) 
 
 p r i n t ( d a t a ) 
 
 
 
 
 
 
 
 
 
 n u m e r i c _ c o l s   =   [ ' A g e ' ,   ' R e s t B P ' ,   ' C h o l ' ,   ' M a x H R ' ,   ' O l d p e a k ' ] 
 
 s c a l e r   =   S t a n d a r d S c a l e r ( ) 
 
 d a t a [ n u m e r i c _ c o l s ]   =   s c a l e r . f i t _ t r a n s f o r m ( d a t a [ n u m e r i c _ c o l s ] ) 
 
 p r i n t ( d a t a ) 
 
 
 
 
 
 X   =   d a t a . d r o p ( ' A H D ' ,   a x i s = 1 ) 
 
 y   =   d a t a [ ' A H D ' ] 
 
 
 
 
 
 
 
 X _ t r a i n ,   X _ t e s t ,   y _ t r a i n ,   y _ t e s t   =   t r a i n _ t e s t _ s p l i t ( 
 
         X ,   y ,   t e s t _ s i z e = 0 . 2 5 ,   r a n d o m _ s t a t e = 1 ) 
 
 p r i n t ( " T r a i n i n g   s e t   s i z e : " ,   X _ t r a i n . s h a p e ) 
 
 p r i n t ( " T e s t i n g   s e t   s i z e : " ,   X _ t e s t . s h a p e ) 