rene/ma_cisco_malware
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

network predicts 2 by 2 classes, refactored threshold to main

Browse Source
This commit is contained in:
René Knaebel
2017-06-30 18:43:50 +02:00
parent 8334e9a84f
commit c972963a19
4 changed files with 24 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/make_csv_dataset.py
View File

@@ -3,8 +3,8 @@
import joblib
import pandas as pd
datafile = joblib.load("/mnt/projekte/pmlcluster/cisco/trainData/multipleTaskLearning/currentData.joblib")
user_flows = datafile["data"]
df = pd.concat(user_flows)
df = joblib.load("/mnt/projekte/pmlcluster/cisco/trainData/multipleTaskLearning/currentData.joblib")
df = df["data"]
df = pd.concat(df)
df.reset_index(inplace=True)
df.to_csv("/tmp/rk/full_dataset.csv.gz", compression="gzip")