Commit c2ef86cb authored by Alessia Marcolini's avatar Alessia Marcolini
Browse files

Remove unused file

parent bf9488e8
from __future__ import division
import numpy as np
import mlpy
import csv
import os.path
import sys
import argparse
parser = argparse.ArgumentParser(description='Compute the Canberra stability indicator given a RANKING file.')
parser.add_argument('RANKFILE', type=str, help='Ranking datafile')
parser.add_argument('METRICSFILE', type=str, help='Metrics datafile (FSTEPS will be read from the 1st column')
parser.add_argument('-f', '--force', action='store_true', help='Force overwriting of output files.')
__author__ = 'Davide Albanese'
if len(sys.argv)==1:
parser.print_help()
sys.exit(1)
args = parser.parse_args()
RANKFILE = args.RANKFILE
METRICSFILE = args.METRICSFILE
overwrite = args.force
OUTFILE = '_'.join(METRICSFILE.split('_')[:-1] + ['stability.txt'])
if os.path.exists(OUTFILE) and not overwrite:
print "Existing output file %s" % (OUTFILE)
sys.exit(2)
stabilityf = open(OUTFILE, 'w')
stability_w = csv.writer(stabilityf, delimiter='\t', lineterminator='\n')
stability_w.writerow(["STEP", "STABILITY"])
RANKING = np.loadtxt(RANKFILE, delimiter='\t', dtype=np.int)
tmp = np.loadtxt(METRICSFILE, delimiter='\t', skiprows=1, dtype=str)
FSTEPS = tmp[1:,0].astype(np.int)
# check dimensions
if RANKING.shape[1] < np.max(FSTEPS):
FSTEPS = np.append( FSTEPS[FSTEPS<RANKING.shape[1]] , RANKING.shape[1] )
STABILITY = []
PR = np.argsort( RANKING )
for ss in FSTEPS:
STABILITY.append( mlpy.canberra_stability(PR, ss) )
for j, s in enumerate(FSTEPS):
stability_w.writerow( [s, STABILITY[j]] )
stabilityf.close()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment