コード例 #1
0
ファイル: views.py プロジェクト: kgantsov/edible_mushroom
def index():
    form = MushroomFeaturesForm()
    result = None

    if form.validate_on_submit():
        tree_data = grab_tree(current_app.config['TREE_DATA_FILE_NAME'])
        data = [
            form.cap_shape.data,
            form.cap_surface.data,
            form.cap_color.data,
            form.bruises.data,
            form.odor.data,
            form.gill_attachment.data,
            form.gill_spacing.data,
            form.gill_size.data,
            form.gill_color.data,
            form.stalk_shape.data,
            form.stalk_root.data,
            form.stalk_surface_above_ring.data,
            form.stalk_surface_below_ring.data,
            form.stalk_color_above_ring.data,
            form.stalk_color_below_ring.data,
            form.veil_type.data,
            form.veil_color.data,
            form.ring_number.data,
            form.ring_type.data,
            form.spore_print_color.data,
            form.population.data,
            form.habitat.data
        ]

        result = classify(tree_data, DATASET_LABELS, data)

    return render_template('main/index.html', form=form, result=result)
コード例 #2
0
    def test_classify(self):
        self.assertEqual(
            classify(
                self.tree_data,
                DATASET_LABELS,
                'x,s,g,f,n,f,w,b,k,t,e,s,s,w,w,p,w,o,e,n,a,g'.split(',')
            ),
            'e'
        )

        self.assertEqual(
            classify(
                self.tree_data,
                DATASET_LABELS,
                'x,y,n,t,p,f,c,n,w,e,e,s,s,w,w,p,w,o,p,n,s,u'.split(',')
            ),
            'p'
        )

        self.assertEqual(
            classify(
                self.tree_data,
                DATASET_LABELS,
                'f,f,g,f,n,f,w,b,n,t,e,f,f,w,w,p,w,o,e,n,a,g'.split(',')
            ),
            'e'
        )

        self.assertEqual(
            classify(
                self.tree_data,
                DATASET_LABELS,
                'f,f,n,f,n,f,w,b,h,t,e,s,s,w,w,p,w,o,e,n,a,g'.split(',')
            ),
            'e'
        )

        self.assertEqual(
            classify(
                self.tree_data,
                DATASET_LABELS,
                'x,f,g,f,f,f,c,b,p,e,b,k,k,b,b,p,w,o,l,h,v,p'.split(',')
            ),
            'p'
        )

        self.assertEqual(
            classify(
                self.tree_data,
                DATASET_LABELS,
                'x,y,g,f,f,f,c,b,p,e,b,k,k,b,b,p,w,o,l,h,y,g'.split(',')
            ),
            'p'
        )