svmlightmodel.html

SVM_light是一种非常流行的支持向量机的java接口

<FONT color="green">181</FONT>          case 0:<a name="line.181"></a>
<FONT color="green">182</FONT>            m_kernel = new LinearKernel();<a name="line.182"></a>
<FONT color="green">183</FONT>            break;<a name="line.183"></a>
<FONT color="green">184</FONT>          case 1:<a name="line.184"></a>
<FONT color="green">185</FONT>            m_kernel = new PolynomialKernel(new LinearKernel(), m_dParam, m_sParam,<a name="line.185"></a>
<FONT color="green">186</FONT>                m_rParam);<a name="line.186"></a>
<FONT color="green">187</FONT>            break;<a name="line.187"></a>
<FONT color="green">188</FONT>          case 2:<a name="line.188"></a>
<FONT color="green">189</FONT>            m_kernel = new RadialBaseKernel(new LinearKernel(), m_gParam);<a name="line.189"></a>
<FONT color="green">190</FONT>            break;<a name="line.190"></a>
<FONT color="green">191</FONT>          case 3:<a name="line.191"></a>
<FONT color="green">192</FONT>            m_kernel = new SigmoidKernel(new LinearKernel(), m_sParam, m_rParam);<a name="line.192"></a>
<FONT color="green">193</FONT>          default:<a name="line.193"></a>
<FONT color="green">194</FONT>            throw new RuntimeException("This type of kernel is not supported!");<a name="line.194"></a>
<FONT color="green">195</FONT>        }<a name="line.195"></a>
<FONT color="green">196</FONT>      }<a name="line.196"></a>
<FONT color="green">197</FONT>    <a name="line.197"></a>
<FONT color="green">198</FONT>      public double classify(FeatureVector v) {<a name="line.198"></a>
<FONT color="green">199</FONT>        double delta = 0;<a name="line.199"></a>
<FONT color="green">200</FONT>        for (int i = 0; i &lt; m_docs.length; i++) {<a name="line.200"></a>
<FONT color="green">201</FONT>          double alpha = m_docs[i].getLabel();<a name="line.201"></a>
<FONT color="green">202</FONT>          if (alpha != 0)<a name="line.202"></a>
<FONT color="green">203</FONT>            delta += alpha * m_kernel.evaluate(m_docs[i], v);<a name="line.203"></a>
<FONT color="green">204</FONT>        }<a name="line.204"></a>
<FONT color="green">205</FONT>        return delta - m_threshold;<a name="line.205"></a>
<FONT color="green">206</FONT>      }<a name="line.206"></a>
<FONT color="green">207</FONT>    <a name="line.207"></a>
<FONT color="green">208</FONT>      public void setThreshold(double threshold) {<a name="line.208"></a>
<FONT color="green">209</FONT>        m_threshold = threshold;<a name="line.209"></a>
<FONT color="green">210</FONT>      }<a name="line.210"></a>
<FONT color="green">211</FONT>    <a name="line.211"></a>
<FONT color="green">212</FONT>      public String toString() {<a name="line.212"></a>
<FONT color="green">213</FONT>        String s = "\n(| Format     : "<a name="line.213"></a>
<FONT color="green">214</FONT>            + m_format<a name="line.214"></a>
<FONT color="green">215</FONT>            + "|\n"<a name="line.215"></a>
<FONT color="green">216</FONT>            + "| KType      : "<a name="line.216"></a>
<FONT color="green">217</FONT>            + m_kType<a name="line.217"></a>
<FONT color="green">218</FONT>            + "\n"<a name="line.218"></a>
<FONT color="green">219</FONT>            + "| dParam     : "<a name="line.219"></a>
<FONT color="green">220</FONT>            + m_dParam<a name="line.220"></a>
<FONT color="green">221</FONT>            + "\n"<a name="line.221"></a>
<FONT color="green">222</FONT>            + "| gParam     : "<a name="line.222"></a>
<FONT color="green">223</FONT>            + m_gParam<a name="line.223"></a>
<FONT color="green">224</FONT>            + "\n"<a name="line.224"></a>
<FONT color="green">225</FONT>            + "| sParam     : "<a name="line.225"></a>
<FONT color="green">226</FONT>            + m_sParam<a name="line.226"></a>
<FONT color="green">227</FONT>            + "\n"<a name="line.227"></a>
<FONT color="green">228</FONT>            + "| rParam     : "<a name="line.228"></a>
<FONT color="green">229</FONT>            + m_rParam<a name="line.229"></a>
<FONT color="green">230</FONT>            + "\n"<a name="line.230"></a>
<FONT color="green">231</FONT>            + "| uParam     : "<a name="line.231"></a>
<FONT color="green">232</FONT>            + m_uParam<a name="line.232"></a>
<FONT color="green">233</FONT>            + "\n"<a name="line.233"></a>
<FONT color="green">234</FONT>            + "| HighFeatIdx   : "<a name="line.234"></a>
<FONT color="green">235</FONT>            + m_highFeatIdx<a name="line.235"></a>
<FONT color="green">236</FONT>            + "\n"<a name="line.236"></a>
<FONT color="green">237</FONT>            + "| trainDocs  : "<a name="line.237"></a>
<FONT color="green">238</FONT>            + m_trainDocs<a name="line.238"></a>
<FONT color="green">239</FONT>            + "\n"<a name="line.239"></a>
<FONT color="green">240</FONT>            + "| numSupVecs : "<a name="line.240"></a>
<FONT color="green">241</FONT>            + m_numSupVecs<a name="line.241"></a>
<FONT color="green">242</FONT>            + "\n"<a name="line.242"></a>
<FONT color="green">243</FONT>            + "| threshold  : "<a name="line.243"></a>
<FONT color="green">244</FONT>            + m_threshold<a name="line.244"></a>
<FONT color="green">245</FONT>            + "\n"<a name="line.245"></a>
<FONT color="green">246</FONT>            + (m_docs != null ? ("#SampleData:" + m_docs.length + "\n|[0]"<a name="line.246"></a>
<FONT color="green">247</FONT>                + m_docs[0] + "|\n" + (0 &lt; m_docs.length ? ("...\n" + "|["<a name="line.247"></a>
<FONT color="green">248</FONT>                + (m_docs.length - 1) + "]" + m_docs[m_docs.length - 1] + "|\n")<a name="line.248"></a>
<FONT color="green">249</FONT>                : "")) : "");<a name="line.249"></a>
<FONT color="green">250</FONT>        return s;<a name="line.250"></a>
<FONT color="green">251</FONT>      }<a name="line.251"></a>
<FONT color="green">252</FONT>    <a name="line.252"></a>
<FONT color="green">253</FONT>      /**<a name="line.253"></a>
<FONT color="green">254</FONT>       * Writes this SVMLightModel to a file. The format is compatible to the<a name="line.254"></a>
<FONT color="green">255</FONT>       * SVM-light model files.<a name="line.255"></a>
<FONT color="green">256</FONT>       */<a name="line.256"></a>
<FONT color="green">257</FONT>      public void writeModelTofile(String path) {<a name="line.257"></a>
<FONT color="green">258</FONT>        File dump = new File(path);<a name="line.258"></a>
<FONT color="green">259</FONT>        if (m_docs != null) {<a name="line.259"></a>
<FONT color="green">260</FONT>          String header = m_format<a name="line.260"></a>
<FONT color="green">261</FONT>              + "\n"<a name="line.261"></a>
<FONT color="green">262</FONT>              + m_kType<a name="line.262"></a>
<FONT color="green">263</FONT>              + " # kernel type\n"<a name="line.263"></a>
<FONT color="green">264</FONT>              + m_dParam<a name="line.264"></a>
<FONT color="green">265</FONT>              + " # kernel parameter -d\n"<a name="line.265"></a>
<FONT color="green">266</FONT>              + m_gParam<a name="line.266"></a>
<FONT color="green">267</FONT>              + " # kernel parameter -g\n"<a name="line.267"></a>
<FONT color="green">268</FONT>              + m_sParam<a name="line.268"></a>
<FONT color="green">269</FONT>              + " # kernel parameter -s\n"<a name="line.269"></a>
<FONT color="green">270</FONT>              + m_rParam<a name="line.270"></a>
<FONT color="green">271</FONT>              + " # kernel parameter -r\n"<a name="line.271"></a>
<FONT color="green">272</FONT>              + m_uParam<a name="line.272"></a>
<FONT color="green">273</FONT>              + "# kernel parameter -u\n"<a name="line.273"></a>
<FONT color="green">274</FONT>              + m_highFeatIdx<a name="line.274"></a>
<FONT color="green">275</FONT>              + " # highest feature index\n"<a name="line.275"></a>
<FONT color="green">276</FONT>              + m_trainDocs<a name="line.276"></a>
<FONT color="green">277</FONT>              + " # number of training documents\n"<a name="line.277"></a>
<FONT color="green">278</FONT>              + m_numSupVecs<a name="line.278"></a>
<FONT color="green">279</FONT>              + " # number of support vectors plus 1\n"<a name="line.279"></a>
<FONT color="green">280</FONT>              + m_threshold<a name="line.280"></a>
<FONT color="green">281</FONT>              + " # threshold b, each following line is a SV (starting with alpha*y)\n";<a name="line.281"></a>
<FONT color="green">282</FONT>    <a name="line.282"></a>
<FONT color="green">283</FONT>          try {<a name="line.283"></a>
<FONT color="green">284</FONT>            BufferedWriter bd = new BufferedWriter(new FileWriter(dump));<a name="line.284"></a>
<FONT color="green">285</FONT>            bd.write(header);<a name="line.285"></a>
<FONT color="green">286</FONT>            for (int i = 0; i &lt; m_docs.length; i++) {<a name="line.286"></a>
<FONT color="green">287</FONT>              bd.write(m_docs[i].toString());<a name="line.287"></a>
<FONT color="green">288</FONT>            }<a name="line.288"></a>
<FONT color="green">289</FONT>            bd.close();<a name="line.289"></a>
<FONT color="green">290</FONT>          } catch (IOException ioe) {<a name="line.290"></a>
<FONT color="green">291</FONT>            ioe.printStackTrace();<a name="line.291"></a>
<FONT color="green">292</FONT>          }<a name="line.292"></a>
<FONT color="green">293</FONT>    <a name="line.293"></a>
<FONT color="green">294</FONT>        } else {<a name="line.294"></a>
<FONT color="green">295</FONT>          System.out.println("Cannot write model file..");<a name="line.295"></a>
<FONT color="green">296</FONT>        }<a name="line.296"></a>
<FONT color="green">297</FONT>      }<a name="line.297"></a>
<FONT color="green">298</FONT>    }<a name="line.298"></a>




























































</PRE>
</BODY>
</HTML>