Israel-Miles

func heapify(heap *[]int, i int) {
	smallest := i
	lChild := 2*i + 1
	rChild := 2*i + 2

	if lChild < len(*heap) && (*heap)[lChild] < (*heap)[smallest] {
		smallest = lChild
	}
	if rChild < len(*heap) && (*heap)[rChild] < (*heap)[smallest] {
		smallest = rChild

Israel-Miles

package main

import (
	"context"
	"log"

	"./pb"

	"google.golang.org/grpc"
)

Israel-Miles

if __name__ == '__main__':
    logging.basicConfig(
        level=logging.INFO,
        format='%(asctime)s - %(levelname)s - %(message)s',
    )
    server = grpc.server(ThreadPoolExecutor())
    add_WineClassifierServiceServicer_to_server(WineClassifierServer(), server)
    port = 8080
    server.add_insecure_port(f'[::]:{port}')
    server.start()

Israel-Miles

def GetWineVariety(self, request, context):
    logging.info(request.review)
    # transform review to document-term matrix
    x_train_counts = self.count_vect.transform([request.review])

    tfidf_transformer = TfidfTransformer()
    x_tfidf = tfidf_transformer.fit_transform(x_train_counts)

    prediction = self.lsvc.predict(x_tfidf)
    for variety, idx in self.top_10_varieties.items():

Israel-Miles

class WineClassifierServer(WineClassifierServiceServicer):
    def __init__(self):
        wine_df = pd.read_csv('data/winemag-data-130k-v2.csv')

        # map top 10 wine varieties to ints in wine_df
        counter = Counter(wine_df['variety'].tolist())
        self.top_10_varieties = {i[0]: idx for idx, i in enumerate(counter.most_common(10))}
        logging.info(self.top_10_varieties)
        wine_df = wine_df[wine_df['variety'].map(lambda x: x in self.top_10_varieties)]

Israel-Miles

# server tools & grpc
from concurrent.futures import ThreadPoolExecutor
from wine_varieties_pb2 import WineReviewResponse
from wine_varieties_pb2_grpc import WineClassifierServiceServicer, add_WineClassifierServiceServicer_to_server
import logging
import grpc

# model construction and selection
from sklearn.svm import LinearSVC
from sklearn.model_selection import train_test_split

Israel-Miles

syntax = "proto3";
package pb;

option go_package = "./pb";

message WineReviewRequest {
  string review = 1;
}

message WineReviewResponse {

Israel-Miles

package main

import (
	"fmt"
	"github.com/sjwhitworth/golearn/base"
	"github.com/sjwhitworth/golearn/evaluation"
	"github.com/sjwhitworth/golearn/knn"
)

func main() {

Israel-Miles

func TrainModel(train dataframe.DataFrame, stream chan gomlBase.TextDatapoint) {
	sentimentMap := map[string]int{
		"0": 0,
		"1": 1,
	}
	for i := 0; i < train.Nrow(); i++ {
		record := train.Subset(i).Records()

		stream <- gomlBase.TextDatapoint{
			X: record[1][0],

Israel-Miles

package main

import (
	"fmt"
	gomlBase "github.com/cdipaolo/goml/base"
	"github.com/cdipaolo/goml/text"
	"github.com/go-gota/gota/dataframe"
	"log"
	"math/rand"
	"os"