#NEWLINE# def predict( self , test_set ) : #NEWLINE# #INDENT# ' Predict the labels for the test set.\n        :param test_set: Testing data. shape(n_test_samples, n_features)\n        :return: labels\n        ' #NEWLINE# with tf . Session ( ) as self . tf_session : #NEWLINE# #INDENT# self . tf_saver . restore ( self . tf_session , self . model_path ) #NEWLINE# return self . model_predictions . eval ( { #NEWLINE# #INDENT# self . input_data : self , #NEWLINE# self . keep_prob : 1 , #NEWLINE# #UNINDENT# } )