col_headers.py

class Header():
	def __init__(self, data=None, malware_col=None):
		malware_col = 'malicious' if malware_col is None else malware_col
		self.training_headers = [
		'vector',
		'memory', 
		'tx_packets',  
		'rx_bytes',
		'swap',
		'rx_packets', 
		'cpu_sys',
		'total_pro',
		'cpu_user', 
		'max_pid',
		'tx_bytes',
		malware_col]
		self.classification_col = self.training_headers.index(malware_col)
		self.vector_col = self.training_headers.index("vector")