Bugfix idf
add probabilistc model