# -*- coding: utf-8 -*-
# ---
# jupyter:
#   jupytext:
#     text_representation:
#       extension: .jl
#       format_name: light
#       format_version: '1.5'
#       jupytext_version: 1.14.4
#   kernelspec:
#     display_name: Julia 1.5.2
#     language: julia
#     name: julia-1.5
# ---

# ### 1.4.1 スパイラル・データセット

include("../dataset/spiral.jl")

using .Spiral
using Plots
gr()

x, t = load_data()

@show size(x)
@show size(t)

N = 100
CLS_NUM = 3
markers = [:circle, :utriangle, :hexagon]
for i in 0:CLS_NUM - 1
    # 1つ目のクラスを描画
    if i == 0
        plot(x[(i * N) + 1:(i + 1) * N, 1], x[(i * N) + 1:(i + 1) * N, 2], st=:scatter, marker = markers[i + 1])
    # 最後のクラスを描画する際に表示する
    elseif i == CLS_NUM - 1　
        display(plot!(x[(i * N) + 1:(i + 1) * N, 1], x[(i * N) + 1:(i + 1) * N, 2], st=:scatter, marker = markers[i + 1]))
    # 表示はせず，上書きとして描画する
    else
        plot!(x[(i * N) + 1:(i + 1) * N, 1], x[(i * N) + 1:(i + 1) * N, 2], st=:scatter, marker = markers[i + 1])
    end
end

StarcoderdataJupyterScriptsDedupFiltered