import category_builder_util as util
def GetArgumentParser():
parser = argparse.ArgumentParser(description='Category Builder')
parser.add_argument('--rho', default=3.0, type=float, help="The rho param")
parser.add_argument('--n',
default=100,
type=int,
help="How many features to use")
parser.add_argument('--cutpaste',
dest='cutpaste',
action='store_true',
help='Prints output in a formay easy to cut-paster')
parser.set_defaults(cutpaste=False)
parser.add_argument('seeds', nargs='+', help="Seeds to expand")
return parser
if __name__ == "__main__":
args = GetArgumentParser().parse_args()
CB = util.CategoryBuilder()
items = CB.ExpandCategory(seeds=args.seeds, rho=args.rho, n=args.n)
if args.cutpaste:
print ', '.join(item[0] for item in items[:50])
else:
for idx, item in enumerate(items[:100]):
print "[%d] %f\t%s" % (idx, item[1], item[0])
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import argparse
import category_builder_util as util
def GetArgumentParser():
parser = argparse.ArgumentParser(description='Category Builder Analogies')
parser.add_argument('--squash',
default=100.0,
type=float,
help="Squash for combining scores")
parser.add_argument('b', help="The B in A:B::C:?")
parser.add_argument('c', help="The C in A:B::C:?")
return parser
if __name__ == "__main__":
args = GetArgumentParser().parse_args()
CB = util.CategoryBuilder(data_dir=".")
items = CB.DoAnalogy(b=args.b, c=args.c, squash=args.squash)
for item in items[:10]:
print(f"{item[1]:5.3f}\t\t{item[0]}")