3  ![!@sddlZddlZddlTddlmZdddddd geZejjejjegeje_d dZ d dZ d dZ d dZ ddd Z dddZdS)N)*)__all__ evaluationssvm_load_model svm_predictsvm_read_problemsvm_save_model svm_trainc Csg}g}xt|D]z}|jdd}t|dkr8|dg7}|\}}i}x.|jD]"}|jd\}} t| |t|<qNW|t|g7}||g7}qW||fS)z svm_read_problem(data_file_name) -> [y, x] Read LIBSVM-format data from data_file_name and return labels y and data instances x. N:)opensplitlenfloatint) Zdata_file_nameZprob_yZprob_xlinelabelfeaturesxieindvalrX/home/kushashwa/Documents/BigVisionLLC/ImageMetrics/Python/libsvm-3.22/python/svmutil.pyrs  cCs.tj|j}|s"td|dSt|}|S)zc svm_load_model(model_file_name) -> model Load a LIBSVM model from model_file_name and return. zcan't open model file %sN)libsvmrencodeprint toPyModel)model_file_namemodelrrrr$s  cCstj|j|dS)ze svm_save_model(model_file_name, model) -> None Save a LIBSVM model to the file model_file_name. N)rrr)rr rrrr1sc Cs$t|t|krtdd}}d}}}}}xnt||D]`\} } | | krX|d7}|| | | | 7}|| 7}|| 7}|| | 7}|| | 7}|| | 7}q@Wt|} d|| } || } yD| |||| |||| |||| |||}Wntd}YnX| | |fS)z evaluations(ty, pv) -> (ACC, MSE, SCC) Calculate accuracy, mean squared error and squared correlation coefficient using the true values (ty) and predicted values (pv). zlen(ty) must equal to len(pv)rr gY@nan)r ValueErrorzipr)typvZ total_correctZ total_errorZsumvZsumyZsumvvZsumyyZsumvyvylACCMSESCCrrrr9s*   DcCsd \}}t|ttfrVt|ttfs(t|||}}}t|}t|||jtkd}n&t|tr||}t|trt|}nt|}|dks|dkrtd|jtkrxT|j D]J}|dj |dj } } |dj dkrt d| dks| |j krt dqW|jdkr|j dkrd|j |_tj|jtj||} | rBt d| |jr|j|j} } t| }tj||| |t|jd| |d| \}}}|jttgkrtd |td ||Std ||Sn tj||}t|}|j |_ |SdS) aP svm_train(y, x [, options]) -> model | ACC | MSE svm_train(prob [, options]) -> model | ACC | MSE svm_train(prob, param) -> model | ACC| MSE Train an SVM model from data (y, x) or an svm_problem prob using 'options' or an svm_parameter param. If '-v' is specified in 'options' (i.e., cross validation) either accuracy (ACC) or mean-squared error (MSE) is returned. options: -s svm_type : set type of SVM (default 0) 0 -- C-SVC (multi-class classification) 1 -- nu-SVC (multi-class classification) 2 -- one-class SVM 3 -- epsilon-SVR (regression) 4 -- nu-SVR (regression) -t kernel_type : set type of kernel function (default 2) 0 -- linear: u'*v 1 -- polynomial: (gamma*u'*v + coef0)^degree 2 -- radial basis function: exp(-gamma*|u-v|^2) 3 -- sigmoid: tanh(gamma*u'*v + coef0) 4 -- precomputed kernel (kernel values in training_set_file) -d degree : set degree in kernel function (default 3) -g gamma : set gamma in kernel function (default 1/num_features) -r coef0 : set coef0 in kernel function (default 0) -c cost : set the parameter C of C-SVC, epsilon-SVR, and nu-SVR (default 1) -n nu : set the parameter nu of nu-SVC, one-class SVM, and nu-SVR (default 0.5) -p epsilon : set the epsilon in loss function of epsilon-SVR (default 0.1) -m cachesize : set cache memory size in MB (default 100) -e epsilon : set tolerance of termination criterion (default 0.001) -h shrinking : whether to use the shrinking heuristics, 0 or 1 (default 1) -b probability_estimates : whether to train a SVC or SVR model for probability estimates, 0 or 1 (default 0) -wi weight : set the parameter C of class i to weight*C, for C-SVC (default 1) -v n: n-fold cross validation mode -q : quiet mode (no outputs) N)isKernelzWrong types for the argumentsrz?Wrong input format: first column must be 0:sample_serial_numberz5Wrong input format: sample_serial_number out of rangeg?z Error: %sz(Cross Validation Mean squared error = %gz5Cross Validation Squared correlation coefficient = %gz Cross Validation Accuracy = %g%%)NN) isinstancelisttupleAssertionError svm_parameter svm_problem kernel_type PRECOMPUTED TypeErrorx_spaceindexvaluer"ngammarsvm_set_print_string_function print_funcsvm_check_parametercross_validationr(nr_foldc_doublesvm_cross_validationrr'svm_type EPSILON_SVRNU_SVRrr r)arg1arg2Zarg3Zprobparamr'xoptionsridxrerr_msgr(r?targetr)r*r+mrrrr VsR%          "    r cCs4dd}d}|j}d}xT|t|krl||dkrH|d7}t||}n||dkrZt}ntd|d7}qW|j}|j} |j} g} g} |r | std|tt gkr|d |j d} t | } x|D]J}t ||j jtkd \}}tj||| }| d | }| |g7} | |g7} qWn| r.|d |tt tfkrDd}n| | dd }t |}xf|D]^}t ||j jtkd \}}tj|||}| dkrdg}n |d |}| |g7} | |g7} qdWt|| \}}}t|}|t tgkr|d||d|n|d|t||d|f| |||f| fS)a svm_predict(y, x, m [, options]) -> (p_labels, p_acc, p_vals) Predict data (y, x) with the SVM model m. options: -b probability_estimates: whether to predict probability estimates, 0 or 1 (default 0); for one-class SVM only 0 is supported. -q : quiet mode (no outputs). The return tuple contains p_labels: a list of predicted labels p_acc: a tuple including accuracy (for classification), mean-squared error, and squared correlation coefficient (for regression). p_vals: a list of decision values or probability estimates (if '-b 1' is specified). If k is the number of classes, for decision values, each element includes results of predicting k(k-1)/2 binary-class SVMs. For probabilities, each element contains k values indicating the probability that the testing instance is in each class. Note that the order of classes here is the same as 'model.label' field in the model structure. cSs t|dS)N)r)srrrinfoszsvm_predict..inforz-br z-qz Wrong optionsz+Model does not support probabiliy estimateszwProb. model for test data: target value = predicted value + z, z: Laplace distribution e^(-|z|/sigma)/(2sigma),sigma=%g)r,Nz@Model supports probability estimates, but disabled in predicton.z$Mean squared error = %g (regression)z1Squared correlation coefficient = %g (regression)z(Accuracy = %g%% (%d/%d) (classification)d)rrr print_nullr" get_svm_typeis_probability_model get_nr_classrDrCget_svr_probabilityr@gen_svm_nodearrayrGr3r4rsvm_predict_probability ONE_CLASSNU_SVCsvm_predict_valuesr)r'rHrMrIrOZpredict_probabilityargvirB is_prob_modelnr_classZ pred_labelsZ pred_valuesZprob_estimatesrrJrvalues nr_classifier dec_valuesr)r*r+r(rrrrsh               )NN)r )ossyssvmrZsvm_allpathdirnameabspath__file__rrrrr rrrrrs    W