xgboost/demo/CLI/yearpredMSD/csv2libsvm.py
2022-11-08 12:59:06 +08:00

13 lines
232 B
Python
Executable File

#!/usr/bin/env python3
import sys
fo = open(sys.argv[2], 'w')
for l in open(sys.argv[1]):
arr = l.split(',')
fo.write('%s' % arr[0])
for i in range(len(arr) - 1):
fo.write(' %d:%s' % (i, arr[i+1]))
fo.close()